Cod sursa(job #2785099)

Utilizator bogdanvladmihaiBogdan Vlad-Mihai bogdanvladmihai Data 17 octombrie 2021 23:04:23
Problema Cowfood Scor 14
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <bits/stdc++.h>

using namespace std;

using ll = long long;

const int mod = 3210121;
const int maxS = (int)1e4;
const int maxN = 30;

int n, s, k;

int c[maxS + maxN + 5][maxN + 5], addM[maxS + maxN + 5];

int m[maxN + 5][maxN + 5];

int st[maxN + 5][maxN + 5];

int bkt(int p, int mask, int rem) {
  if (rem < 0) {
    return 0;
  }
  if (p == n) {
    int pw = ((__builtin_popcount(mask) % 2 == 0) ? 1 : -1);
    return pw * addM[rem];
  }
  for (int i = 0; i < k; i++) {
    st[p + 1][i] = st[p][i];
  }
  int x = bkt(p + 1, mask, rem);
  int newRem = s;
  for (int i = 0; i < k; i++) {
    st[p + 1][i] = max(m[p][i], st[p][i]);
    newRem -= st[p + 1][i];
  }
  return (x + bkt(p + 1, (mask | (1 << p)), newRem)) % mod;
}

int main() {
  freopen("cowfood.in", "r", stdin);
  freopen("cowfood.out", "w", stdout);
  cin >> n >> s >> k;
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < k; j++) {
      cin >> m[i][j];
    }
  }
  for (int i = 0; i <= k; i++) {
    c[0][i] = 1;
  }
  for (int i = 0; i <= s; i++) {
    c[i][0] = 1;
  }
  for (int i = 1; i <= s; i++) {
    for (int j = 1; j <= k; j++) {
      c[i][j] = ((ll)c[i - 1][j] + c[i][j - 1]) % mod;
    }
  }
  for (int i = 0; i <= s; i++) {
    addM[i] = c[i][k];
  }
  cout << (bkt(0, 0, s) - k * s - 1 + mod) % mod << "\n";
  return 0;
}