#P11358. [eJOI2023] Tree Infection
[eJOI2023] Tree Infection
题目描述
给定一棵 个点的有根树 ,保证根结点的编号为 。
树上的一个点 会被选中,距离 不超过 的点会被感染,两个点的距离定义为其树上路径的边数。
一对点是可达的当且仅当它们都没有被感染,且其树上路径经过被感染的点数不超过 。
对于 计算选中 的可达点对数量。
输入格式
第一行输入三个整数 。
第二行输入 个整数 ,代表每个点的父亲。
输出格式
输出 行,每行一个整数,第 行的整数代表选择 的答案。
13 2 2
1 2 3 4 3 6 6 8 2 10 11 1
16
4
15
55
66
36
66
55
66
45
55
66
66
3 0 1
1 2
1
1
1
提示
【样例解释】
的树如图所示。
所有可达点对为 ,, 和 。
注意 不可达,因为 已经被感染; 不可达,因为路径上有三个被感染的点 。
【数据范围】
本题采用捆绑测试。
- Subtask 1(20 pts):。
- Subtask 2(14 pts):。
- Subtask 3(15 pts):。
- Subtask 4(10 pts):。
- Subtask 5(16 pts):。
- Subtask 6(25 pts):无特殊限制。
对于 的数据,保证 ,,,。