CF1278F

CF932E 其实是差不多的捏

p=1m,q=1p,那么枚举第一张是王牌的次数,有如下式子:

ni=1(ni)piqniik

后面那个 ik 可以展开为第二类斯特林数:

=ni=1(ni)piqnikj=1{kj}ij_

=kj=1{kj}ni=1(ni)piqniij_

考虑化简

ni=1(ni)piqniij_

=ni=j(ni)piqni(ij)j!

=j!(nj)ni=j(njij)piqni

=nj_nji=0(nji)pi+jqnji

=nj_pjnji=0(nji)piqnji

注意到后面那个就是 (p+q)nj 用二项式定理展开的形式,而 p+q=1

所以

=nj_pj

那么答案就是

kj=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;
}
posted @ 2022-10-25 07:24  Kobe303  阅读(12)  评论(0编辑  收藏  举报