Desserts(组合数学 + 思维)

题目链接：Desserts

题目大意：

有 n 种糖果，第 i 种糖果共有 ai 个，问把所有糖果分给 k 支队伍，使得每支队伍不会拥有相同类型糖果的方案数。保证 $\sum ai$ ≤ $10^5$。

解题思路：

首先，考虑暴力做法，对于第 i 种糖果分给 k 支队伍的方法有 $C_k^{ai}$ 种， 则ans = ${\prod }_{i=1}^n{C}_k^{ai}$。可以看出这样复杂度为O(nm),会超时，所以我们应该想办法优化一下。

由$\sum ai$ ≤ $10^5$ 可以推出不同的 ai 最多有$\sqrt{\sum ai}$种,所以直接枚举不同的 ai ，然后相同的 ai 快速幂贡献在一起即可。这样复杂度就成了O(m$\sqrt{\sum ai}$ logn)

AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
const int mod = 998244353;
int fact[N], infact[N], a[N];
int n, m;
unordered_map<int, int> mp;
int qmi(int a, int k, int p)
{
    int res = 1;
    while (k)
    {
        if (k & 1) res = (ll)res * a % p;
        a = (ll)a * a % p;
        k >>= 1;
    }
    return res;
}
int main()
{
    fact[0] = infact[0] = 1;
    for (int i = 1; i <= 1e5; i++)
    {
        fact[i] = (ll)fact[i - 1] * i % mod;
        infact[i] = (ll)infact[i - 1] * qmi(i, mod - 2, mod) % mod;
    }
    scanf("%d%d", &n, &m);
    int maxx = 0;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        mp[a[i]]++;
        maxx = max(maxx, a[i]);
    }
    for (int i = 1; i <= m; i++)
    {
        if (i < maxx) puts("0");
        else
        {
            ll ans = 1;
            for (auto &u : mp)
            {
                int x = u.first, y = u.second;
                ans = ans * qmi((ll)fact[i] * infact[x] % mod * infact[i - x] % mod, y, mod) % mod;
            }
            printf("%lld\n", ans);
        }
    }
	return 0;
}