Rainbow Roads （链式前向星，树上深搜）

Rainbow Roads

题目

 题目大意

找出点的个数，满足从这个点开始的每一对相邻边，没有相同的颜色

思路

        一开始读完题，考虑到两种情况，一种是一个点出发的存在相邻的边颜色相同，一种是一个点出发的两个节点有相同的颜色。刚开始往树上dp想了，但状态转移的中间量实在是不会存，就想着先把两种情况的其中一种实现出来。

       于是就遍历每一个点，和它们的第一级下属节点，把有重复颜色的标出来。实现完发现前一种情况也迎刃而解了，直接对每一个标记的点继续向下深搜就行，深搜出来的每一个点都打上标记。最后对没打上标记的点进行统计就好了。

出现问题

1.vector存边被卡了，第11个样例的时候T了，所以就换成了链式前向星

2.链式前向星有几个变量忘记开双倍大小.......debug了好久

#include <bits/stdc++.h>
using namespace std;
const int N = 5e4+100;

int n;
vector< pair<int,int> >ve[N];
int p[N];
int aa[N];

int cnt=0;
int e[2*N];
int ne[2*N];
int h[2*N];
int w[2*N];
void add(int a,int b,int c) {//链式前向星存边
	e[cnt]=b;
	ne[cnt]=h[a];
	h[a]=cnt;
	w[cnt]=c;
	cnt++;
}
vector<int>ans;
void dfs(int u,int fa) {//对点进行深搜
	if(p[u]>5)return;//2开始应该就可以了
	p[u]++;
	for(int i=h[u]; i!=-1; i=ne[i]) {
		int b=e[i];
		if(b!=fa) {
			dfs(b,u);
		}
	}
}
int main() {
	memset(h,-1,sizeof(h));
	int f=1;
	cin>>n;
	for(int i=1; i<n; i++) {//存边
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	for(int a=1; a<=n; a++) {//判子节点重颜色
		for(int i=h[a]; i!=-1; i=ne[i]) {
			int c=w[i];
			aa[c]++;
		}
		for(int i=h[a]; i!=-1; i=ne[i]) {
			int b=e[i],c=w[i];
			if(aa[c]>1) {
				dfs(b,a);
			}
		}
		for(int i=h[a]; i!=-1; i=ne[i]) {//这么删降低点复杂度
			int c=w[i];
			aa[c]--;
		}

	}
	for(int i=1; i<=n; i++) {//存答案 
		if(p[i]==0)ans.push_back(i);
	}
	cout<<ans.size()<<"\n";
	for(int i=0; i<ans.size(); i++) {
		printf("%d\n",ans[i]);
	}
	return 0;
}