Cod sursa(job #2937079)

Utilizator Catalinu23Gavrila Catalin Catalinu23 Data 9 noiembrie 2022 21:00:23
Problema Diamant Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <bits/stdc++.h>
#define MOD 10000
using namespace std;

ifstream fin("diamant.in");
ofstream fout("diamant.out");

int n, m, x, len;
int dp[2][95000], a[405];

int main()
{
    fin >> n >> m >> x;
    if (abs(x) > n*(n+1)/2*m*(m+1)/2)
        fout<<0<<"\n";
    else
    {
        for (int i=1; i<=n; i++)
        {
            for (int j=1; j<=m; j++)
            {
                a[++len] = i*j;
            }
        }
        x += 45000;
        dp[0][45000] = 1;
        int L = 0;
        for (int i = 1; i <= len; i++)
        {
            L = 1 - L;
            for (int j = 0; j <= 90000; j++)
            {
                dp[L][j] = dp[1-L][j];
                if (j >= a[i])
                {
                    dp[L][j] += dp[1-L][j-a[i]];
                }
                dp[L][j] += dp[1-L][j+a[i]];
                dp[L][j] = dp[L][j] % MOD;
            }
        }
        fout << dp[L][x];
    }
    return 0;
}