Pagini recente » Cod sursa (job #753268) | Cod sursa (job #2084712) | Cod sursa (job #2933663) | Cod sursa (job #1334155) | Cod sursa (job #986549)
Cod sursa(job #986549)
#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;
}