hihcoder 1117 战争年代

Problem

有一棵有n个节点的树，每一轮会等概率随机一个$[0,d_i]$内的数$u$，然后与$x_i$距离不超过$u$的点都会被涂成一个新的颜色，相邻的同颜色的点视为同一联通块，问m轮后联通块的期望个数。
$1\le n\le 2000$

Solution

联通块个数=两边的点的颜色不同的边的个数+1
于是我们可以对每条边维护一下两边的点颜色不同的概率，假设有边$(u,v)$且$dis[x_i][u]<dis[x_i][v]$，那么一次的更新是$f'(u,v)=f(u,v)\times \frac{dis[x_i][u]}{d_i+1}+\frac{1}{d_i+1}$
最后将所有的概率加起来就好了。