A. Timofey and a tree_思维

A. Timofey and a tree_思维

题目大意：

给一颗无根树，现要选出一个节点作为根，使得其所有子树的节点颜色分别相同。问是否可能。

思路和代码：

一开始看到只有单组数据，试了试暴力做法，即将每个点都试做根，跑一遍搜索。但是这样做会T。

bool dfs(int now , int pre , vct<vct<int>> &eg , vct<int> &c){
	int f = 1 ;
	for(int nxt : eg[now]){
		if(nxt == pre) continue ;
		if(c[now] != c[nxt]) return 0 ;
		if(!dfs(nxt , now , eg , c)) return 0 ;
	}
	return 1 ; 
}

void solve(){
	int n ;
	cin >> n ;
	vct<int> c(n + 1 , 0) ;
	vct<vct<int> > eg(n + 1) ;
	rep(i , 2 , n){
		int u , v ; cin >> u >> v ;
		eg[u].pb(v) ;
		eg[v].pb(u) ;
	} 
	
	rep(i , 1 , n) cin >> c[i] ;
	 
	rep(i , 1 , n){
		int f = 1 ;
		for(auto rt : eg[i]){
			if(!dfs(rt , i , eg , c))f = 0 ;
			if(!f) break ;
		}
		
		if(f){
			cout << "YES\n" << i << "\n" ;
			return ;
		}
	} 
	cout << "NO\n" ;
	
}//code_by_tyrii 

于是想其他做法，发现要满足所有子树颜色分别相同，就只能把颜色不同的边全部插在大根上。

所以代码也是比较简单的，如下。

void solve2(){
	int n ;
	cin >> n ;
	
	vct<int> c(n + 1 , 0) ;
	vct<pii> e(n , {0 , 0}) ;
	vct<int> cnt(n + 1 , 0) ;
	int all = 0 ;
	
	rep(i , 0 , n - 2)
	cin >> e[i].fi >> e[i].se ;
	
	rep(i , 1 , n) cin >> c[i] ;
	
	rep(i , 0 , n - 1){
		int u = e[i].fi ; 
		int v = e[i].se ; 
		if(c[u] != c[v]){
			all ++ ;
			cnt[u] ++ , cnt[v] ++ ;
		}
	}
	rep(i , 1 , n){
		if(cnt[i] == all){
			cout << "YES\n" << i << "\n" ;
			return ;
		}
	}cout << "NO\n" ;
	
}