CF1278F
与 CF932E 其实是差不多的捏
设 p=1m,q=1−p,那么枚举第一张是王牌的次数,有如下式子:
n∑i=1(ni)piqn−iik
后面那个 ik 可以展开为第二类斯特林数:
=n∑i=1(ni)piqn−ik∑j=1{kj}ij_
=k∑j=1{kj}n∑i=1(ni)piqn−iij_
考虑化简
n∑i=1(ni)piqn−iij_
=n∑i=j(ni)piqn−i(ij)j!
=j!(nj)n∑i=j(n−ji−j)piqn−i
=nj_n−j∑i=0(n−ji)pi+jqn−j−i
=nj_pjn−j∑i=0(n−ji)piqn−j−i
注意到后面那个就是 (p+q)n−j 用二项式定理展开的形式,而 p+q=1。
所以
=nj_pj
那么答案就是
k∑j=1{kj}nj_pj
O(k2) 预处理斯特林数啥的就好了。
Code:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5005, mod = 998244353;
int n, m, k, p;
int ans;
int S[N][N], mul[N], f[N];
int qpow(int x, int y) {
int res = 1;
while (y) {
if (y & 1) res = 1ll * res * x % mod;
x = 1ll * x * x % mod;
y >>= 1;
}
return res;
}
void init() {
S[0][0] = 1;
for (int i = 1; i <= k; ++i)
for (int j = 1; j <= i; ++j)
S[i][j] = (S[i - 1][j - 1] + 1ll * S[i - 1][j] * j % mod) % mod;
mul[0] = 1;
for (int i = 1; i <= k; ++i) mul[i] = 1ll * mul[i - 1] * p % mod;
f[0] = 1;
for (int i = 1; i <= k; ++i) f[i] = 1ll * f[i - 1] * (n - i + 1) % mod;
}
int main() {
scanf("%d%d%d", &n, &m, &k), p = qpow(m, mod - 2);
init();
for (int i = 1; i <= min(n, k); ++i) {
int tmp = 1ll * S[k][i] * mul[i] % mod * f[i] % mod;
ans = (ans + tmp) % mod;
}
printf("%d", ans);
return 0;
}