题目描述
给定一棵 n n n个点的树,在 k k k个位置上存在蝴蝶,我们需要给 n − 1 n-1 n−1条边定向,如果一条边的起点有蝴蝶且终点没有蝴蝶,那么蝴蝶将被移动到终点,我们会按照给定边的顺序移动,问最终所有蝴蝶的树上距离的和的期望,答案除于 k ( k − 1 ) 2 \frac{k(k-1)}{2} 2k(k−1),对 998244353 998244353 998244353取模
k ≤ n ≤ 300000 k\le n\le 300000 k≤n≤300000
题解
首先考虑第一个问题,我们怎么求 k k k个点的距离和,可以随便钦定一个点为根,求出 s i z u siz_u sizu表示子树 u u u中的指定结点个数,答案显然就是 ∑ s i z u ( k − s i z u ) \sum siz_u(k-siz_u) ∑sizu