Codeforces Round #646 (Div. 2) E. Tree Shuffling（树上贪心）

题目链接

传送门

题目大意

一颗大小为$n$，$1$为根的树，对于每个点，有：花费$a_i$，原二进制位$b_i$，目标二进制位$c_i$。现在要完成的任务就是：进行多次的操作，使得$b_i$等于$c_i$。

操作：可以从结点$u$为根的子树中，选出任意$k$个结点，然后可以自己排列这$k$个结点的$b_i$的值（即安排这些值放的点的位置），代价是$k\ast a_u$。

问：要找到最小的代价总和，使得$b_i=c_i$ $(1\le i\le n)$。

题目分析

关键思路：把问题转化为：只与子树有关，与祖先结点无关的贪心问题。

具体思路：首先可以忽略$b_i=c_i$的点，接下来剩下两种点：

种类1：$b_i=0$，$c_i=1$
种类2：$b_i=1$，$c_i=0$

合法的条件：种类1的点的数量等于种类2的点的数量。原因：只有交换种类1和种类2的点的$b_i$，才能使$b_i=c_i$。

为了使问题转化为子树上的贪心，与祖先结点无关，可以进行这个操作：$a_i=min(a_i,a_p)$，其中，结点$p$为结点$i$的父结点。

现在，我们思考贪心的方向：如果从根开始，把所有种类1的点，和种类2的点进行交换，那么，付出的代价是最大的。均摊代价的话，每个点花费的代价：$a_1$。如果对于一些结点，在它们所属的同一子树进行交换，是不是可以使这些点的均摊代价小于等于$a_1$？（核心：$a_i=min(a_i,a_p)$）

所以，为了最小化每个点花费的代价，我们从叶子结点开始往上解决问题：统计子树中，种类1和种类2的点的数量，能交换多少就尽量地交换。最后这个子树只有种类1，或者只有种类2的点，然后把这些剩余的点的数量传给父结点解决。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
const int maxn = 2e5 + 5;

LL a[maxn], b[maxn], c[maxn];
LL ans;
vector<int> G[maxn];

P dfs(int u, int last) {
	a[u] = min(a[u], a[last]);

	P par(0, 0);
	if (b[u] == 0 && c[u] == 1) par.first++;
	if (b[u] == 1 && c[u] == 0) par.second++;

	for (int i = 0; i < G[u].size(); i++) {
		int to = G[u][i];
		if (to != last) {
			P t = dfs(to, u);
			par.first += t.first;
			par.second += t.second;
		}
	}

	LL cnt = min(par.first, par.second);
	ans += 2 * cnt * a[u];
	par.first -= cnt;
	par.second -= cnt;

	return par;
}

int main() {
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		scanf("%lld %lld %lld", &a[i], &b[i], &c[i]);
	}
	for (int i = 1; i <= n - 1; i++) {
		int u, v;
		scanf("%d %d", &u, &v);
		G[u].push_back(v);
		G[v].push_back(u);
	}

	a[0] = 1e9 + 5;
	P par = dfs(1, 0);
	if (par.first || par.second) printf("-1\n");
	else printf("%lld\n", ans);

	return 0;
}