Cod sursa(job #3219239)

Utilizator Robert_MitriRobert Mitri Robert_Mitri Data 30 martie 2024 16:12:42
Problema Cowfood Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.36 kb
#include <fstream>
using namespace std;
ifstream fin("cowfood.in");
ofstream fout("cowfood.out");

const int mod = 3210121;

const int kmax = 30;
const int smax = 10000;

long long n,k,s;
bool bk[kmax + 5];

long long a[kmax + 5][kmax + 5];
long long d[smax + 5][kmax + 5];
long long sol;

void bt(int pas,int nrset = 0)
{
    if(pas==n)
    {
        if(nrset==0)
            return;
        long long smx = 0;
        for(int j = 1;j<=k;j++)
        {
            long long s=0;
            for(int i=1;i<=n;i++)
                if(bk[i])
                    s=max(s,a[i][j]);
            smx+=s;
        }
        if(nrset%2)
            sol =(sol - d[s-smx][k+1]+mod)%mod;
        else
            sol =(sol + d[s-smx][k+1])%mod;
        return;
    }
    bt(pas+1,nrset);
    bk[pas+1]=1;
    bt(pas+1,nrset+1);
    bk[pas+1]=0;
}



int main()
{
    fin>>k>>s>>n;

    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++)
            fin>>a[i][j];


    for(int j = 0;j<=k+1;j++)
        d[0][j]=1;
    for(int i = 1;i<=s+1;i++)
        for(int j = 0;j<=k+1;j++)
            if(j<=1)
                d[i][j] = 1;
            else
                d[i][j]=(d[i-1][j]+d[i][j-1])%mod;

    sol =(sol+ d[s][k+1] - s*k -1);
    while(sol < 0)
        sol+=mod;
    bt(0);
    while(sol < 0)
        sol+=mod;
    fout<<sol;
}