Cod sursa(job #3298784)

Utilizator anca.gdDumitru Anca Gabriela anca.gd Data 1 iunie 2025 17:28:08
Problema Diamant Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.52 kb
#include <fstream>
using namespace std;
int valori[403], negativ[50003],pozitiv[50003],  vfneg[50003],vfpoz[50003];
int n,m,X;
int mod=10000;
int main()
{
    ifstream fin("diamant.in");
    ofstream fout("diamant.out");
    fin>>n>>m>>X;
    if(X>45000) {
        fout<<0;
        return 0;
    }
    if(X<-45000) {
        fout<<0;
        return 0;
    }
    int indx=0;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m;j++)
            valori[++indx]=i*j;
    negativ[1]=1; pozitiv[1]=1;
    for (int k=2; k<=indx; k++){
        int initial=pozitiv[0];
        //+1
        for(int i=50000; i>0; i--)
            if(pozitiv[i]){
                pozitiv[i+valori[k]]+=pozitiv[i];
                pozitiv[i+valori[k]]%=mod;
                vfpoz[i+valori[k]]=pozitiv[i];
            }
        for(int i=1; i<50000; i++)
            if(negativ[i]){
                int where=-i+valori[k];
                if(where>=0){
                    pozitiv[where]+=negativ[i];
                    pozitiv[where]%=mod;
                    vfpoz[where]=negativ[i];
                }
                else{
                    negativ[-where]+=negativ[i];
                    negativ[-where]%=mod;
                    vfneg[-where]=negativ[i];
                }
            }
        //-1
        for (int i=50000; i>0; i--)
            if(negativ[i]){
                int where=i+valori[k];
                negativ[where]+=negativ[i];
                negativ[where]-=vfneg[i];
                negativ[where]+=mod;
                negativ[where]%=mod;
            }
        for(int i=1; i<50000; i++)
            if(pozitiv[i]){
                int where=i-valori[k];
                if(where>=0){
                    pozitiv[where]+=pozitiv[i];
                    pozitiv[where]-=vfpoz[i];
                    pozitiv[where]+=mod;
                    pozitiv[where]%=mod;
                }
                else{
                    negativ[-where]+=pozitiv[i];
                    negativ[-where]-=vfpoz[i];
                    negativ[-where]+=mod;
                    negativ[-where]%=mod;
                }
            }
        for (int i=1; i<50000; i++)
            vfpoz[i]=0, vfneg[i]=0;
        pozitiv[valori[k]]+=initial;
        pozitiv[valori[k]]%=mod;
        negativ[valori[k]]+=initial;
        negativ[valori[k]]%=mod;
        if(!initial) pozitiv[valori[k]]++, negativ[valori[k]]++;
    }
    if(X<0) fout<<negativ[-X];
        else fout<<pozitiv[X];
    fout<<'\n';
    return 0;
}