Codeforces Round #646 (Div. 2) E（贪心，bfs）

Codeforces Round #646 (Div. 2) E

题目大意： 给一棵树，每个节点有三个权值 A,B,C, (B,C为0或1)，每次你可以花费 A[u] *k的代价让A子树中的任意 k 个节点交换彼此的 B ，问让所有节点的 B=C 至少花费多少代价。

思路： b[i]=1时 代表1 0，b[i]=-1代表0 1， 等于0就是相同的，不需要交换。然后bfs，从下往上（从上往下递归）开始计算，b[i]就代表这颗树还又多少个不能被交换的。

代码：

#include
using namespace std;
 
#define LL long long
LL ans; int x, y, n, a[200005], b[200005];
vector<int> g[200005];
 
void dfs(int v, int par) {
	int tot = abs(b[v]);
	for(int u : g[v]) {
		if(u == par) continue;
		a[u] = min(a[v], a[u]);
		dfs(u, v); b[v] += b[u];
		tot += abs(b[u]);
	}
	ans += 1LL * (tot - abs(b[v])) * a[v];
}
 
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n;
	for(int i=1;i<=n;i++) {
		cin >> a[i] >> x >> y;
		b[i] = x - y;
	}
	while(--n) {
		cin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	dfs(1, 0);
	cout << (b[1] ? -1 : ans);
}