Cod sursa(job #2950548)

Utilizator Matei_MunteanuMunteanu Matei Ioan Matei_Munteanu Data 4 decembrie 2022 00:13:11
Problema Cowfood Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.04 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fin("cowfood.in");
ofstream fout("cowfood.out");
const int MOD = 3210121;

int n, s, k;
int E[24][34];
int rez;

int putere(int a, int p)
{
    int rez = 1;
    while (p)
    {
        if (p % 2 == 0)
        {
            a = (1LL * a * a) % MOD;
            p /= 2;
        }
        else
        {
            rez = (1LL * rez * a) % MOD;
            p--;
        }
    }
    return rez;
}

int comb(int n, int k)
{
    int factn = 1;
    for (int i = 1; i <= n; i++)
    {
        factn = (1LL * factn * i) % MOD;
    }
    int invk = 1;
    for (int i = 1; i <= k; i++)
    {
        invk = (1LL * invk * putere(i, MOD - 2)) % MOD;
    }
    for (int i = 1; i <= (n - k); i++)
    {
        invk = (1LL * invk * putere(i, MOD - 2)) % MOD;
    }
    return (1LL * factn * invk) % MOD;
}
int main()
{
    fin >> n >> s >> k;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= k; j++)
        {
            fin >> E[i][j];
        }
    }
    int nr_total = (comb(s + k, k) - (k * s) % MOD - 1) % MOD;
    for (int conf = 1; conf < (1 << n); conf++)
    {
        int maxi[34];
        int nr = 0;
        memset(maxi, -1, sizeof(maxi));
        for (int i = 0; i < n; i++)
        {
            if (conf & (1 << i))
            {
                nr++;
                for (int j = 1; j <= k; j++)
                {
                    maxi[j] = max(maxi[j], E[i + 1][j]);
                }
            }
        }
        int sum = 0;
        for (int i = 1; i <= k; i++)
        {
            sum += maxi[i];
        }
        int dif = s - sum;
        if (dif > 0)
        {
            int comun = comb(dif + k, k);
            if (nr % 2 == 1)
            {
                rez = (rez + comun) % MOD;
            }
            else
            {
                rez = (rez + comun) % MOD;
            }
        }
    }
    fout << (nr_total - rez) % MOD;
    fin.close();
    fout.close();
    return 0;
}