Cod sursă (job #285782)

#include <iostream>
#include <fstream>
using namespace std;
ifstream f ("2sah.in");
ofstream g ("2sah.out");
int a[9000][9000]={0};
int main()
{
    int p,n,k,i,j;
    int s=0;
    f>>p>>n>>k;
    if(p==1)
    {
        for(i=1;i<=k;i++)
        {
            a[i][((2*n+1)/2)+2-i]=a[i][((2*n+1)/2)+i]=1;
            for(j=((2*n+1)/2)+2-i+1;j<((2*n+1)/2)+i;j++)
            {
                a[i][j]=a[i-1][j-1]+a[i-1][j]+a[i-1][j+1];
            }
        }
        for(j=((2*n+1)/2)+2-i+1;j<((2*n+1)/2)+i;j++)
        {
            s+=(a[k][j])%100003;
        }
    g<<s%100003;
    }
    if(p==2)
    {
        for(i=1;i<=n+1;i++)
        {
            a[i][((2*n+1)/2)+2-i]=a[i][((2*n+1)/2)+i]=1;
            for(j=((2*n+1)/2)+2-i+1;j<((2*n+1)/2)+i;j++)
            {
                a[i][j]=a[i-1][j-1]+a[i-1][j]+a[i-1][j+1];
            }
        }
        i=1;
        while(i<=n+1&&k<=(2*n)+1)
        {
            s+=(a[i][k])%100003;
            i++;
            k+=2;
        }
    g<<s%100003;
    }
}