Cod sursa(job #893296)

Utilizator timicsIoana Tamas timics Data 26 februarie 2013 14:38:35
Problema Diamant Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.99 kb
#include<stdio.h>
#include<algorithm>
using namespace std;
int X,M,N,s[2][90000],t[401];
int main()
{
    freopen("diamant.in","r",stdin);
    freopen("diamant.out","w",stdout);
    scanf("%d%d%d",&N,&M,&X);
    if((X<-N*M*(M+1)*(N+1)/4)||(X>N*M*(M+1)*(N+1)/4))
    {
        printf("%d",0);
        return 0;
    }
    for(int i=1;i<=M*N;++i)
    {
        t[i]=t[i-1]+2*((i-1)/M +1)*((i-1)%M+1);
    }
    X=X+N*M*(M+1)*(N+1)/4;
    s[0][0]=1;
    s[0][1]=1;
    s[0][2]=1;
    for(int i=2;i<=N*M;++i)
    {
        for(int j=0;j<=min(t[i],X);++j)
        {
            s[1][j]=s[0][j];
            if(j>=((i-1)/M +1)*((i-1)%M+1))
                s[1][j]=(s[1][j]+s[0][j-((i-1)/M +1)*((i-1)%M+1)])%10000;
            if(j>=2*((i-1)/M +1)*((i-1)%M+1))
                s[1][j]=(s[1][j]+s[0][j-2*((i-1)/M +1)*((i-1)%M+1)])%10000;
        }
        for(int j=0;j<=min(t[i],X);++j)
        {
            s[0][j]=s[1][j];
        }
    }
    printf("%d",s[0][X]);
    return 0;
}