Cod sursă (job #285758)

#include <iostream>
#include <fstream>
using namespace std;
ifstream f ("2sah.in");
ofstream g ("2sah.out");
int main()
{
    int p,n,k,i,j,s=0;
    f>>p>>n>>k;
    int a[700][700]={0};
    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];
        }
    g<<s;
    }
    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];
            i++;
            k+=2;
        }
    g<<s;
    }
}