P9346 无可奈何花落去 题解

P9346 无可奈何花落去 题解

这个期望第一眼看上去是困难的。然而发现 E=Px,贡献 x 是可枚举的,于是转化为了一个求概率的问题。这个概率同样难以计算,然而发现状态的个数是有限的,对于选取 x 条边断掉它的总方案数就是 (n1x),那么直接求方案数就可以了。

想到了这些转化这题基本就做完了。直接暴力树上背包设 dpi,j,0/1/2 表示 i 点为根的子树内,断掉了 j 条边,i0/1/2 种儿子相连的方案数,最终算出 fi=dpj(n1x)。注意到我们求的断 i 跳边的情况事实上包含了跳 0i1 条边的情况,于是做的时候减去 0i1fi 即可。具体实现的时候可以前缀和累加去算,但其实不前缀和也不影响复杂度。

代码:

#include <bits/stdc++.h>
#define N 5005
#define int long long
#define mod 985661441
using namespace std;
int n;
vector<int>v[N];
signed dp[N][N][3];
int tmp[N][3];
void add(int &x, int y) {
	x = (x + y) % mod;
}
int siz[N];
void dfs(int x) {
	siz[x] = 1;
	dp[x][0][0] = 1;
	for (int y : v[x]) {
		dfs(y);
		for (int j = 0; j < siz[x]; j++)
			for (int k = 0; k < siz[y]; k++) {
				int p = (0ll + dp[y][k][0] + dp[y][k][1] + dp[y][k][2]) % mod, q = (0ll + dp[y][k][0] + dp[y][k][1]) % mod;
				add(tmp[j + k + 1][0], dp[x][j][0] * p % mod);
				add(tmp[j + k + 1][1], dp[x][j][1] * p % mod);
				add(tmp[j + k][1], dp[x][j][0] * q % mod);
				add(tmp[j + k + 1][2], dp[x][j][2] * p % mod);
				add(tmp[j + k][2], dp[x][j][1] * q % mod);
			}
		for (int j = 0; j < N; j++)
			for (int k = 0; k < 3; k++) 
				dp[x][j][k] = tmp[j][k], tmp[j][k] = 0;
		siz[x] += siz[y];
	}
}

int fac[N], inv[N];
int qpow(int x, int y) {
	int ans = 1;
	while (y) {
		if (y & 1) ans = ans * x % mod;
		x = x * x % mod;
		y >>= 1;
	}
	return ans;
}
int C(int n, int m) {
	return fac[n] * inv[m] % mod * inv[n - m] % mod;
}
int res;
int f[N];
signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	fac[0] = 1;
	for (int i = 1; i < N; i++) fac[i] = fac[i - 1] * i % mod;
	inv[N - 1] = qpow(fac[N - 1], mod - 2);
	for (int i = N - 2; ~i; --i) inv[i] = inv[i + 1] * (i + 1) % mod;
	cin >> n;
	for (int i = 2; i <= n; i++) {
		int x;
		cin >> x;
		v[x].push_back(i);
	}
	dfs(1);
	for (int i = 0; i < n; i++) {
		add(f[i], 0ll + dp[1][i][0] + dp[1][i][1] + dp[1][i][2]);
		f[i] = f[i] * qpow(C(n - 1, i) , mod - 2) % mod;
		add(res, (f[i] - f[i - 1] + mod) % mod * i % mod);
	}
	cout << res << "\n";
	return 0;
}
posted @   长安19路  阅读(7)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示