Cod sursa(job #986549)

Utilizator Impaler_009Mihai Nitu Impaler_009 Data 19 august 2013 01:22:19
Problema Cowfood Scor 42
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.71 kb
#include <fstream>
#define maxs 10040
#define ll long long

using namespace std;

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

ll k,s,n;
ll f[maxs],crs[maxs],aux;
ll m[31][21],temp[21];

ll x,nkinv,kinv,NK,K,N,t,S;
ll mod =3210121;

void inverse (ll a, ll b, ll &x, ll &y)
{
    if (b==0)
    {
        x=1;
        y=0;
        return;
    }
    ll xp,yp;
    inverse (b,a%b,xp,yp);
    x = yp;
    y = xp - yp*(a/b);
}

int subsets ()
{
    t = (1<<n)-1;

    for (int i=1; i<=t; ++i)
    {
        int nr=0;

        for (int j=0; j<n; j++)
        {
            if ((i>>j)&1)
            {
                ++nr;
                for (int h=1; h<=k; ++h)
                   temp[h] = max (temp[h],m[j+1][h]);
            }
        }

        int ts=0;
        for (int h=1; h<=k; ++h) ts+=temp[h];
        ts=(s-ts);
        if (ts>=0)
        {
            if (nr&1) {S -= crs[ts]; if (S<0) S+=mod;}
            else {S += crs[ts]; if (S>=mod) S-=mod;}
        }
        for (int h=1; h<=k; ++h) temp[h]=0;
    }
}

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

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

    f[0]=1;
    for (int i=1; i<=s+k; ++i) f[i] = (f[i-1]*i) % mod;

    crs[0]=1;

    NK = f[k-1];

    inverse (mod,NK,x,nkinv);

    if (nkinv<=0) nkinv = mod + nkinv%mod;

    for (int i=1; i<=s; ++i)
    {
        N = f[i+k-1];
        K = f[i];

        inverse (mod,K,x,kinv);
        if (kinv<=0) kinv = mod + kinv%mod;

        crs[i]=crs[i-1]+(((N*kinv)%mod)*nkinv)%mod;
        if (crs[i]>=mod) crs[i]-=mod;
    }

    S += crs[s]-1;
    S = (S-s*k)%mod;

    subsets ();

    fout<<S;
}