Cod sursa(job #467108)

Utilizator R.A.RFMI Romila Remus Arthur R.A.R Data 28 iunie 2010 11:36:36
Problema Pod Scor 5
Compilator cpp Status done
Runda Stelele Informaticii 2010, clasele X-XII, Ziua 2 Marime 0.8 kb
#include <fstream>
#define MOD 9901

using namespace std;

ifstream in("pod.in");
ofstream out("pod.out");

long long N,M,K,i,x,p,s_sol;
int main()
{
    in>>N>>M>>K;
    if(M==0)
    {
        x = N/K;//nr x
        p = N-x*K;//nr p
        if(p==0)
        {
            s_sol++;
            x-=1;
            p+=K;
        }
        if(p==1)
        {
            s_sol+=(x+1);
            x-=1;
            p+=K;
        }
        if(x==0)
        {
            s_sol++;
        }
        s_sol%=MOD;
        while(x!=1)
        {
            s_sol+=(x+p)*(x+p-1)/2;
            x--;
            p+=K;
            if(s_sol>MOD) s_sol%=MOD;
        }
        if(x==1)
        {
            s_sol+=(p+2);
        }
    }
    s_sol%=MOD;
    out<<s_sol;
    return 0;
}