Cod sursa(job #2810014)

Utilizator grecu_tudorGrecu Tudor grecu_tudor Data 28 noiembrie 2021 11:16:39
Problema Cowfood Scor 8
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <fstream>
#include <iostream>
#include <vector>

using namespace std;

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

const int MAXS = 10005;
const int MAXK = 35;
const int MAXN = 25;
const int MOD = 3210121;

int cnt[MAXN][MAXK];
int comb[MAXS + MAXK][MAXK];
int k, s, n;

void precalc() {
  for ( int i = 1; i < MAXS + MAXK; ++i ) {
	comb[i][1] = i;
    for ( int j = 2; j < MAXK; ++j ) {
      comb[i][j] = (comb[i - 1][j] + comb[i - 1][j - 1]) % MOD;
    }
  }
}

int sum[MAXN];
int sp, h, bad;
vector<int> a, b;

void pinex( int pos ) {
  if ( pos == n ) {
	if ( h == 0 ) return;
	if ( h & 1 ) bad = (bad + sum[s - sp]) % MOD;
	else bad = (bad - sum[s - sp]) % MOD;
	return;
  }
  pinex( pos + 1 );
  ++h;
  for ( int i = 0; i < k; ++i ) {
	b[i] = a[i];
	sp -= a[i];
	a[i] = max(a[i], cnt[pos][i]);
	sp += a[i];
  }
  pinex( pos + 1 );
  for ( int i = 0; i < k; ++i ) {
	sp -= a[i];
	a[i] = b[i];
	sp += a[i];
  }
  --h;
}

int main() {
  fin >> k >> s >> n;
  for ( int i = 0; i < n; ++i ) {
	for ( int j = 0; j < k; ++j ) {
	  fin >> cnt[i][j];
	}
  }
  precalc();
  sum[0] = 1;
  for ( int i = 1; i <= s; ++i ) {
    sum[i] = (sum[i-1] + comb[i + k - 1][k - 1]) % MOD;
  }
  for ( int i = 0; i < k; ++i ) {  
	a.push_back(0);
    b.push_back(0);
  }
  pinex( 0 );
  bad = (bad + 1 + s * k) % MOD;
  fout << (sum[s] - bad + MOD) % MOD;
  fin.close();
  fout.close();
  return 0;
}