[蓝桥杯 2023 省 A] 颜色平衡树
题目描述
给定一棵树,结点由 $1$ 至 $n$ 编号,其中结点 $1$ 是树根。树的每个点有一个颜色 $C_i$。
如果一棵树中存在的每种颜色的结点个数都相同,则我们称它是一棵颜色平衡树。
求出这棵树中有多少个子树是颜色平衡树。
输入输出格式
输入格式
输入的第一行包含一个整数 $n$,表示树的结点数。
接下来 $n$ 行,每行包含两个整数 $C_i,F_i$,用一个空格分隔,表示第 $i$ 个结点的颜色和父亲结点编号。
特别地,输入数据保证 $F_1$ 为 $0$,也即 $1$ 号点没有父亲结点。保证输入数据是一棵树。
输出格式
输出一行包含一个整数表示答案。
输入输出样例
输入样例 #1
6
2 0
2 1
1 2
3 3
3 4
1 4
输出样例 #1
4
说明
#### 【样例说明】
编号为 $1,3,5,6$ 的 $4$ 个结点对应的子树为颜色平衡树。
#### 【评测用例规模与约定】
对于 $30 \%$ 的评测用例,$n \leq 200$,$C_i \leq 200$;
对于 $60 \%$ 的评测用例,$n \leq 5000$,$C_i \leq 5000$;
对于所有评测用例,$1 \leq n \leq 200000$,$1 \leq C_i \leq 200000$,$0 \leq F_i<i$。