码迷,mamicode.com
首页 > 其他好文 > 详细

[Dwango Programming Contest 6th C] Cookie Distribution

时间:2020-02-26 01:04:57      阅读:80      评论:0      收藏:0      [点我收藏+]

标签:task   一个   getchar   amp   lin   air   coder   class   方案   

传送门

组合意义,妙啊(弱菜 swk 不会数数实锤了

考虑 \(\prod c_i\) 的组合意义,它代表每个孩子在他得到的饼干中选择一个拿出来,最终的方案数。

因此得到了一个几乎显然的 dp:

\(f_{i, j}\) 为前 \(i\) 天,有 \(j\) 个孩子已经 “选择了” 他的饼干的方案数。转移时枚举新增了 \(u\) 个孩子,则转移系数是一系列组合数,详见代码。

#include <bits/stdc++.h>
#define R register
#define mp make_pair
#define ll long long
#define pii pair<int, int>
using namespace std;
const int N = 1100, K = 22, mod = 1e9 + 7;

int n, k, a[K];
ll f[K][N], comb[N][N];

template <class T>
inline void read(T &x) {
    x = 0;
    char ch = getchar(), w = 0;
    while (!isdigit(ch)) w = (ch == '-'), ch = getchar();
    while (isdigit(ch)) x = (x << 1) + (x << 3) + (ch ^ 48), ch = getchar();
    x = w ? -x : x;
    return;
}

inline ll addMod(ll a, ll b) {
    return (a += b) >= mod ? a - mod : a;
}

int main() {
    read(n), read(k);
    for (R int i = 1; i <= k; ++i) read(a[i]);
    for (R int i = 0; i <= n; ++i) {
        comb[i][0] = 1;
        for (R int j = 1; j <= i; ++j)
            comb[i][j] = addMod(comb[i - 1][j - 1], comb[i - 1][j]);
    }
    f[0][0] = 1;
    for (R int i = 1; i <= k; ++i)
        for (R int j = 0; j <= n; ++j)
            for (R int l = max(0, j - a[i]); l <= j; ++l)
                f[i][j] = (f[i][j] + f[i - 1][l] * comb[n - l][j - l] % mod * comb[n - j + l][a[i] - j + l]) % mod;
    cout << f[k][n] << endl;
    return 0;
}

[Dwango Programming Contest 6th C] Cookie Distribution

标签:task   一个   getchar   amp   lin   air   coder   class   方案   

原文地址:https://www.cnblogs.com/suwakow/p/12364818.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!