Cod sursa(job #1413002)

Utilizator savulescustefanSavulescu Stefan savulescustefan Data 1 aprilie 2015 18:01:04
Problema Sandokan Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.46 kb
#include <cstdio>

using namespace std;
int n,k,s,i,p;
int main()
{
    freopen ("sandokan.in","r",stdin);
    freopen ("sandokan.out","w",stdout);
    scanf ("%d %d", &n, &k);
    if (k==2)
    printf ("%d", n-1);
    else if (k==3)
    printf ("%d", (((n-2)*(n-1))/2)%2000003);
    else
    {
    for (i=(n-k+1);i>=1;i--)
    {
        p=i*(i+1);
        p/=2;
        s+=p;
        s%=2000003;
    }
    printf ("%d", s);
    }
    return 0;
}