#P13962. [2024多校联盟省选模拟]人赢
[2024多校联盟省选模拟]人赢
题目描述
昨天表白失败,属实有点破防。
小 Z 正在表白小 W,他们之间的关系可以看作是一棵有根树,根是 。每个点要么属于小 Z,要么属于小 W,可以用一个长度为 的 序列表示:第 个数为 表示点 属于小 Z,为 表示属于小 W。
定义小 Z 向小 W 表白次数为:点对 的个数,满足 是 的祖先,并且点 属于小 Z、点 属于小 W。
为了尽可能减少表白次数,你可以翻转至多 个节点(把某个属于小 Z 的点变成小 W,或把小 W 变成小 Z),使表白次数最小。
你需要回答 的所有情况的答案。
输入格式
第一行一个正整数 。
第二行一个长度为 的 序列。
接下来 行每行两个正整数 ,表示树上的一条边 。
输出格式
一行 个数,第 个数表示 时的答案。
8
1 1 1 0 1 0 1 0
7 3
1 4
5 1
7 5
6 7
2 6
2 8
8 4 1 0 0 0 0 0 0
10
0 1 0 0 1 1 1 0 0 0
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
14 9 6 2 0 0 0 0 0 0 0
样例解释
对于样例 1:
- 时,将 号点变成 。
- 时,将 号点变成 。
- 时,将 号点变成 。
数据范围与提示
| 测试点编号 | 特殊性质 | |
|---|---|---|
| 20 | ||
| 500 | ||
| 8000 | A | |
| B | ||
- 特殊性质 A:保证 序列最多含有 个 。
- 特殊性质 B:保证树是一条以 为端点的链。