Pagini recente » Borderou de evaluare (job #351662) | Monitorul de evaluare | Cod sursa (job #3307271) | Monitorul de evaluare | Cod sursa (job #3340501)
#include <bits/stdc++.h>
#define ll long long
#define ld double
using namespace std;
ifstream fin("cowfood.in");
ofstream fout("cowfood.out");
const int NMAX = 20;
const int KMAX = 30;
const int SMAX = 2e4;
const int MOD = 3210121;
int k, s, n, answer;
int a[NMAX + 1][KMAX + 1];
int maxi[KMAX + 1];
int fact[SMAX + 1], invfact[SMAX + 1];
int power(int a, int b) {
int rez = 1;
while(b) {
if(b % 2 == 1) {
rez = (ll) rez * a % MOD;
}
a = (ll) a * a % MOD;
b /= 2;
}
return rez;
}
int combinari(int n, int k) {
return (ll) fact[n] * invfact[k] % MOD * invfact[n - k] % MOD;
}
int main() {
fin >> k >> s >> n;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= k; j++) {
fin >> a[i][j];
}
}
fact[0] = 1;
for(int i = 1; i <= 2 * s; i++) {
fact[i] = (ll) fact[i - 1] * i % MOD;
}
invfact[2 * s] = power(fact[2 * s], MOD - 2);
for(int i = 2 * s - 1; i >= 0; i--) {
invfact[i] = (ll) invfact[i + 1] * (i + 1) % MOD;
}
for(int mask = 0; mask < (1 << n); mask++) {
fill(maxi + 1, maxi + k + 1, 0);
for(int i = 1; i <= n; i++) {
if(mask >> (i - 1) & 1) {
for(int j = 1; j <= k; j++) {
maxi[j] = max(maxi[j], a[i][j]);
}
}
}
int sum_here = 0;
for(int j = 1; j <= k; j++) {
sum_here += maxi[j];
}
if(sum_here <= s) {
int sign = (__builtin_popcount(mask) % 2 == 1 ? -1 : 1);
answer = ((ll) answer + sign * combinari(s - sum_here + k, k) % MOD + MOD) % MOD;
}
}
fout << ((answer - 1 - s * k) % MOD + MOD) % MOD << '\n';
return 0;
}