#1629. 「雅礼集训 2018 Day7」C
「雅礼集训 2018 Day7」C
题目描述
给定一棵 个点的树,树上每个点初始有一个 或 的数字。
考虑这样一个过程:
- 等概率随机选择一个点作为起点
- 等概率随机选择一个新点并沿着树上的路径移动过去,最后反转这个新点上的数字(注意只反转这个新点上的数字而非经过的所有点的数字)
- 如果此时整棵树上的所有数字相同,则过程结束;否则回到步骤
求出期望的移动距离,对 取模。
输入格式
第一行一个整数 。
第二行一个长度为 的 串,表示每个点的初始数字。
接下来 行,其中第 行一个整数 表示一条连接 和 的边。
输出格式
一行一个整数表示答案。
样例 1
2
01
1
500000004
- 初始点为 ,选择的新点为 ,过程结束,距离为
- 初始点为 ,选择的新点为 ,过程结束,距离为
- 初始点为 ,选择的新点为 ,过程结束,距离为
- 初始点为 ,选择的新点为 ,过程结束,距离为
答案为 ,对 取模等于 。
3
001
1
1
638888896
一种可能的方案如下:
- 从 号点出发,选择了 号点自己,这一步的移动步数为 , 号点数字变为
- 选择 号点,这一步的移动步数为 , 号点数字变为
- 选择 号点,这一步的移动步数为 , 号点数字变为
- 此时所有点均变为 ,过程结束,总移动步数为
最终答案为 ,对 取模等于 。
数据范围与提示
对于全部数据, ,保证初始局面不满足终止条件。
- 存在 的数据, 。
- 存在 的数据, 。
- 存在 的数据, 。
- 存在 的数据, 。