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<bits/stdc++.h>
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);
}