Cod sursa(job #1670821)

Utilizator george_stelianChichirim George george_stelian Data 1 aprilie 2016 01:34:46
Problema Cowfood Scor 18
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <cstdio>
#include <algorithm>

using namespace std;

const int mod=3210121;
int v[25][35],fact[10100],comb[10100],v1[35],aux[25][35],n,k,s,sol;

int rid_put(int a,int n)
{
    int p=1;
    for(int i=1;i<=n;i<<=1)
    {
        if(n&i) p=(1LL*p*a)%mod;
        a=(1LL*a*a)%mod;
    }
    return p;
}

void back(int poz,int nr)
{
    if(poz==n+1)
    {
        if(nr==0) return;
        int a=s+k;
        for(int i=1;i<=k;i++) a-=v1[i];
        if(nr&1) sol+=comb[a];
        else sol-=comb[a];
        if(sol>=mod) sol-=mod;
        if(sol<mod) sol+=mod;
    }
    else
    {
        back(poz+1,nr);
        for(int i=1;i<=k;i++) aux[poz][i]=v1[i];
        for(int i=1;i<=k;i++) v1[i]=max(v1[i],v[poz][i]);
        back(poz+1,nr+1);
        for(int i=1;i<=k;i++) v1[i]=aux[poz][i];
    }
}

int main()
{
    freopen("cowfood.in", "r", stdin);
    freopen("cowfood.out", "w", stdout);
    scanf("%d%d%d",&k,&s,&n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=k;j++) scanf("%d",&v[i][j]);
    fact[0]=1;
    for(int i=1;i<=s+k;i++) fact[i]=(1LL*fact[i-1]*i)%mod;
    for(int i=k;i<=s+k;i++)
        comb[i]=(1LL*fact[i]*rid_put((1LL*fact[k]*fact[i-k])%mod,mod-2))%mod;
    back(1,0);
    sol=(comb[s]-sol+mod)%mod;
    printf("%d",sol);
    return 0;
}