AT_dwacon6th_prelims_c Cookie Distribution 题解

组合意义保平安。

思路#

发现 的贡献不好统计。

我们可以考虑 的组合意义。

容易发现:

ci=j=1ci1

那么依照分配律,我们发现这个东西的组合意义是每个人从获得的饼干中选一个出来的方案。

这样就会变好统计很多。

dpi,j 为前 i 天,有 j 个人已经选出了代表的饼干。

那么:

dpi,j=k=0jdpi1,k(nkj)(njaij)

时间复杂度:O(n2)

Code#

#include <bits/stdc++.h>
using namespace std;

#define int long long

const int mod = 1e9 + 7;

int n, k;
int f[1010];
int g[1010];
int c[1010][1010];

signed main() {
  cin >> n >> k;
  f[0] = 1;
  for (int i = 0; i <= n; i++) c[i][0] = 1;
  for (int i = 1; i <= n; i++)
    for (int j = 1; j <= i; j++)
      c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % mod;
  for (int i = 1; i <= k; i++) {
    int a;
    cin >> a;
    memset(g, 0, sizeof g);
    for (int j = 0; j <= n; j++) {
      for (int k = 0; k <= a; k++) {
        if (j + k > n) break;
        g[j + k] = (g[j + k] + f[j] * c[n - j][k] % mod * c[n - k][a - k]) % mod;
      }
    }
    memcpy(f, g, sizeof g);
  }
  cout << f[n] << "\n";
}

作者:JiaY19

出处:https://www.cnblogs.com/JiaY19/p/18403294

版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。

posted @   JiaY19  阅读(6)  评论(0编辑  收藏  举报
more_horiz
keyboard_arrow_up light_mode palette
选择主题
menu
点击右上角即可分享
微信分享提示