#P9233. [蓝桥杯 2023 省 A] 颜色平衡树

[蓝桥杯 2023 省 A] 颜色平衡树

题目描述

给定一棵树,结点由 11nn 编号,其中结点 11 是树根。树的每个点有一个颜色 CiC_i

如果一棵树中存在的每种颜色的结点个数都相同,则我们称它是一棵颜色平衡树。

求出这棵树中有多少个子树是颜色平衡树。

输入格式

输入的第一行包含一个整数 nn,表示树的结点数。

接下来 nn 行,每行包含两个整数 Ci,FiC_i,F_i,用一个空格分隔,表示第 ii 个结点的颜色和父亲结点编号。

特别地,输入数据保证 F1F_100,也即 11 号点没有父亲结点。保证输入数据是一棵树。

输出格式

输出一行包含一个整数表示答案。

6
2 0
2 1
1 2
3 3
3 4
1 4
4

提示

【样例说明】

编号为 1,3,5,61,3,5,644 个结点对应的子树为颜色平衡树。

【评测用例规模与约定】

对于 30%30 \% 的评测用例,n200n \leq 200Ci200C_i \leq 200

对于 60%60 \% 的评测用例,n5000n \leq 5000Ci5000C_i \leq 5000

对于所有评测用例,1n2000001 \leq n \leq 2000001Ci2000001 \leq C_i \leq 2000000Fi<i0 \leq F_i<i