Cod sursa(job #873343)

Utilizator vlad_DVlad Dumitriu vlad_D Data 7 februarie 2013 08:50:42
Problema Cowfood Scor 18
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <cstdio>
#include <algorithm>

using namespace std;
int dp[31][10000];
int exp[31][31];
int mod = 3210121;
int cate(int N, int S) {
  if (S < 0) return 0;
  if (N == 1) return S + 1;
  if (dp[N][S]) return dp[N][S];
  dp[N][S] = 0;
  for (int i = 0; i <= S; ++i)
    dp[N][S] = (dp[N][S] + cate(N-1, S - i)) % mod;
  return dp[N][S];
}
int main() {
  freopen("cowfood.in", "r", stdin);
  freopen("cowfood.out", "w", stdout);
  int K, S, N;
  scanf("%d %d %d", &K, &S, &N);
  for (int i = 1; i <= N; ++i)
    for (int j = 1; j <= K; ++j) scanf("%d", &exp[j][i]);
  
  int ret = cate(K, S);
  for (int mask = 1; mask < (1 << N); ++mask) {
    int semn = 1;
    if (__builtin_popcount(mask) % 2) semn = -1;
    int T = S;
    for (int i = 1; i <= K; ++i) {
      int mx = 0;
      for (int j = 0; j < N; ++j) if (mask & (1<<j)) mx = max(mx, exp[i][j+1]);
      T -= mx;
    }
    ret += semn * cate(K, T);
    while (ret < 0) ret += mod;
    ret %= mod;
  }
  // 1 singur nul.
  ret -= cate(K - 1, S - K + 1) * K;
  ret--;
  while (ret < 0) ret += mod;
  printf("%d\n", ret);
  return 0;
}