Cod sursa(job #773511)

Utilizator vendettaSalajan Razvan vendetta Data 1 august 2012 21:48:12
Problema Diamant Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.7 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream f("diamant.in");
ofstream g("diamant.out");

#define Xmax 45000
#define MOD 10000

int prec[2*Xmax];
int acum[2*Xmax];

#define prec (prec + Xmax)
#define acum (acum + Xmax)

int n, m, X;

int main(){

    //X -ul poate fi maxim 45000 sau -45000 (aproximativ);cel mai rau caz e de genul 1*1+..+1*m + 2*1+...+2*m + ... m*1+..m*n(pun doar 1 sau 0)

/*
    int s = 0;
    for(int i=1; i<=20; i++){
        for(int j=1; j<=20; j++){
            s += i*j;
        }
    }
    cout << s << "\n";
*/

    f >> n >> m >> X;
    if (X < -Xmax || X > Xmax){
        g << 0 << "\n";
        return 0;
    }
    //iau fiecare pozitie si pun -1 0 sau 1; punanad valoarea asta obtin -i*j, 0, sau i*j si cu astea valoare continui tot ce am obtinut pana acum

    for(int i=-Xmax+1; i<=Xmax-1; i++) prec[i] = 0, acum[i] = 0;

    prec[0] = 1;
    acum[0] = 0;

    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){

            //for(int k=-Xmax+1; k<=Xmax-1; k++) acum[k] = prec[k]%MOD;//pun 0;

            for(int k=-Xmax+1; k<=Xmax-1; k++){
                if ( k+i*j <= Xmax-1){
                    acum[k+i*j] += prec[k];//pun -1;
                    //if (acum[k+i*j] >= MOD)
                        acum[k+i*j] %= MOD;
                }
                if (k-i*j >= -Xmax+1){
                    acum[k-i*j] += prec[k];//pun 1;
                    //if (acum[k-i*j] >= MOD)
                        acum[k-i*j] %= MOD;
                }
            }

            for(int k=-Xmax+1;  k<= Xmax-1; k++){
                prec[k] = acum[k] % MOD;
            }
        }
    }

    g << acum[X] << "\n";

}