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