小红的树上赋值（easy）（DFS&贪心）--牛客周赛 Round 41-E

解析

考虑 $DFS$ 和贪心，首先从叶结点向上处理，并且由于绝对值最大，所以赋值只有 $0,1,-1$ 三种情况，并且贪心的赋值 $1$ 。

对于当前结点的子树，其权值之和大于 $1$ 的情况，我们遍历其孩子结点，将其权值为 $1$ 并且颜色为白的结点权值改为 $-1$ 直到 $sum$ 小于 $1$ 为止。

对于当前结点，我们已知其子树的权值之和，然后根据当前结点颜色进行选择将其赋值为 $0$ 还是其他值即可。

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define inf 1e18
const int mod=1e9+7;
const int N=1e5+5;
int n,l,r,ans[N];
char a[N];
vector<int>e[N];
int dfs(int u,int fa){
	int sum=0;
	for(auto it:e[u]){
		if(it!=fa) sum+=dfs(it,u);
	}
	if(sum>1){
		for(auto it:e[u]){
			if(sum<2) break;
			if(it!=fa&&ans[it]==1&&a[it]=='W'){
				sum-=2;
				ans[it]=-1;	
			}
		}
	}
	if(a[u]=='R'){
		ans[u]-=sum;
		sum=0;
	}
	else{
		ans[u]=1;
		sum+=1;
	}
	return sum;
}
void solve(){
	cin>>n>>l>>r;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(1,-1);
	for(int i=1;i<=n;i++){
		cout<<ans[i]<<" ";
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int tt=1;
//	cin>>tt;
	while(tt--) solve();
	return 0;
}