BZOJ5322: [JXOI2018]排序問題

傳送門
不難看出指望就是 \(\frac{(n+m)!}{\prod_{v=1}^{max}(cnt_v!)}\)，\(cnt_v\) 表示 \(v\) 這個數出現的次數。
貪心就是直接把 \(m\) 個數字每次選擇一個 \(cnt\) 最小的加入，使得最後 \([l,r]\) 內每一個數字出現的次數儘可能平均。
直接按照數字出現的次數排序，從大到小枚舉平均的次數是多少便可。

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

const int maxn(2e5 + 5);
const int mod(998244353);

inline void Inc(int &x, const int y) {
    x = x + y >= mod ? x + y - mod : x + y;
}

inline int Pow(ll x, int y) {
    ll ret = 1;
    for (; y; y >>= 1, x = x * x % mod)
        if (y & 1) ret = ret * x % mod;
    return ret;
}

int n, m, l, r, a[maxn], val[maxn], cnt[maxn], tot, ans;
int fac[(int)1e7 + maxn];

inline void Solve() {
    int i, len, facn;
    scanf("%d%d%d%d", &n, &m, &l, &r), len = r - l + 1;
    for (i = 1; i <= n; ++i) scanf("%d", &a[i]);
    sort(a + 1, a + n + 1), facn = fac[n + m];
    for (i = 1, tot = 0; i <= n; ++i) {
        if (a[i] != val[tot]) val[++tot] = a[i], cnt[tot] = 0;
        ++cnt[tot];
    }
    n = tot, ans = 1, tot = 0;
    for (i = 1; i <= n; ++i)
        if (val[i] < l || val[i] > r) ans = (ll)ans * fac[cnt[i]] % mod;
        else cnt[++tot] = cnt[i], m += cnt[i];
    sort(cnt + 1, cnt + tot + 1);
    for (i = tot; ~i; --i) {
        if (m / len >= cnt[i]) {
            ans = (ll)Pow(fac[m / len + 1], m % len) * ans % mod * Pow(fac[m / len], len - m % len) % mod;
            break;
        }
        m -= cnt[i], --len, ans = (ll)ans * fac[cnt[i]] % mod;
    }
    ans = (ll)Pow(ans, mod - 2) * facn % mod;
    printf("%d\n", ans);
}

int main() {
    int i, mx = 1e7 + 2e5;
    for (i = fac[0] = 1; i <= mx; ++i) fac[i] = (ll)fac[i - 1] * i % mod;
    scanf("%d", &mx);
    while (mx--) Solve();
    return 0;
}