【蓝桥杯】颜色平衡树（暴力解法）

 【蓝桥杯 2023 省 A】 颜色平衡树

【题目描述】     

        给定一棵树，结点由 1 至 n 编号，其中结点 1 是树根。树的每个点有一个颜色 Ci。

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

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

【输入格式】

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

        接下来 n行，每行包含两个整数 Ci，Fi，用一个空格分隔，表示第 i个结点的颜色和父亲结点编号。

        特别地，输入数据保证 F1为 0，也即 1号点没有父亲结点。保证输入数据是一棵树。

【输出格式】        

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

样例         
6
2 0
2 1
1 2
3 3
3 4
1 4
 

样例输出 
4
 

        借鉴网上大佬思路，利用dfs遍历子结点，样例通过80%。如果利用启发式合并可以拿满分，但囿于笔者算法掌握程度有限，只得借鉴暴力做法。这是我在网上看到比较好理解的暴力做法，复盘如下：

        数组color[N]用于记录每个节点颜色，下标即为节点编号。每个节点有一种颜色，所以开个一维数组即可。

        vector<int> son[N]用于记录每个节点的孩子节点，下标即为父节点编号。向量（Vector）是一个封装了动态大小数组的顺序容器，在本题中可以简单理解成一个二维动态数组。通过son[f].push_back(i)记录f节点子结点编号；son[f].size()返回元素数，即子结点的个数，可在后续dfs中遍历子结点时使用；要访问son[f]中所存的子结点，可以像访问二维数组一样直接访问son[f][i]。

        将输入读取后从1号节点开始遍历，dfs(1,i)中传入的i是unordered_map<int,int>类型，记录本节点所存储的颜色种类和出现次数（向上递归，包含子树的颜色种类）【unordered_map则是一个无序的键值对容器，允许通过键（key）快速查找对应的值（value）】

        在dfs函数中，如果son[f].size()==0，子结点数为0，当前遍历点为叶节点，记录本节点颜色数并将结果加1（叶节点一定是颜色平衡树）

        unordered_map<int,int> it；it[color[x]]++; 用以记录自己的颜色  first记录颜色 second记录颜色出现的次数，向上传递至父节点。

        用for循环对每个子结点进行遍历，for循环后的代码是叶子节点向上递归。pre代表当前颜色数（修改前），初始为0；f为状态位，初始为1。如果pre被修改且与新加入的节点任一颜色数不同，则颜色一定不平衡；反之平衡，结果加1。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<vector>
using namespace std;

const int N=2e5+10;
vector<int> son[N];
int color[N];
int n,ans=0;

void dfs(int x, unordered_map<int,int> &mp){
	if(son[x].size()==0){
		mp[color[x]]++;
		ans++;
		return;
		
	}//if
	
	unordered_map<int,int> it;
	it[color[x]]++;//记录自己的颜色  first记录颜色 second记录颜色出现的次数 
	
	for(int i=0;i<son[x].size();i++){
		dfs(son[x][i],it);//遍历子结点	
		
	}//for
	
	int pre=0,f=1;
	
	for(auto&i:it){
		mp[i.first]+=i.second;
		if(pre&&pre!=i.second){
			f=0;
		}
		pre=i.second;
		
	}//for
	
	if(f==1) ans++;
		
}//dfs

int main(){
	scanf("%d",&n);
	int f=0;
	
	for(int i=1;i<=n;i++){
		scanf("%d%d",&color[i],&f);
		son[f].push_back(i);
	}//for
	
	unordered_map<int,int> i;
	dfs(1,i);
	
	printf("%d",ans);
	
	return 0;
		
}//main