#Z1103. good point
Description
给你一棵树，N个点，每个点都有其颜色,树的根结点为1号点
我们称一个点X是一个good point，仅当：
从顶点1到顶点X的路径上，没有别的顶点，其颜色与X的颜色是一样的
Format
Input
一行给出数字N
接下来1行，给出N个顶点的颜色，其值<=1e5
接下来N-1行描述这个树
N<=1e5
Output
一行一个数字，从小到大输出所有good point
Samples
输入数据 1
6
2 7 1 8 2 8
1 2
3 6
3 2
4 3
2 5
Copy
输出数据 1
1
2
3
4
6思路
用桶记录颜色出现了几次，如果
ton[col[x]] == 0则加入答案数组即可。
代码
#include <bits/stdc++.h>
using namespace std;
int col[10000001],n,u,v,ton[1000001],ans[1000001],al;
vector<int> vec[100001];
void dfs(int x,int fa)
{
if(ton[col[x]] == 0) ans[++al] = x;
ton[col[x]]++;
for(int i = 0;i < vec[x].size();i++)
if(vec[x][i] != fa)
dfs(vec[x][i],x);
ton[col[x]]--;
}
int main()
{
cin>>n;
for(int i = 1;i <= n;i++) cin>>col[i];
for(int i = 1;i < n;i++)
{
cin>>u>>v;
vec[u].push_back(v);
vec[v].push_back(u);
}
dfs(1,0);
sort(ans + 1,ans + 1 + al);
for(int i = 1;i <= al;i++) cout<<ans[i]<<endl;
return 0;
}