UVA1205 Color a Tree （贪心 + 优先队列）

Description

给定一棵有 $n$ 个节点的树，每个节点有一个权值 $w_i$，你需要从前往后一个一个地把点安放进数组 $a$ 中。$a$ 需要满足若点 $a_i$ 为 $a_j$ 的父亲，则 $i$ 必须小于 $j$ 。一种安放方案的权值为 ${\sum }_{i=1}^n{w}_{a[i]}\times i$，请找出权值最小的方案并输出最小权值。

$1\le n\le 1000$。

Solution

如果没有父节点排在节点之前的限制，就可以按照权值从大到小排序的顺序选。对于一个权值极大的点，如果没有限制的话我们越早放这个点越好，但是有限制的话，就必须将这个点与他的父亲绑定在一起，一旦他的父亲放了，就紧挨着他的父亲放进去。

先不考虑如何绑定节点。如果在保持这个绑定关系同时，其他节点不作限制，我们应该如何排列呢？ 我们假设绑定在一起的节点是 $a$ 和 $b$ 。现有另外一个节点 $x$，我们观察两种排列 $xab$，$abx$ 对最终的计算结果有什么影响，假设第一个数的位置在序列中是 $i$ ，这两个排列带来的权值分别是：
$$va{l}_1=w_x\times i+w_a\times (i+1)+w_b\times (i+2)$$

$$va{l}_2=w_a\times i+w_b\times (i+1)+w_x\times (i+2)$$

显然谁的 $val$ 小谁就选谁。

$$va{l}_2-va{l}_1=2\times w_x-(w_a+w_b)$$

如果结果 $>0$，那么就选前者，否则选后者。这相当于比较 $2\times w_x$ 与 $w_a+w_b$ 谁大，左右同时除二，比较的是 $\frac{w_x}{1}$ 与 $\frac{w_a+w_b}{2}$。

发现这个东西好像是平均数？也就是说，设一个序列 $w_1$ 有 $n$ 个节点，另一个序列 $w_2$ 的序列有 $m$ 个节点，我们可以比较 $\frac{{\sum }_{i=1}^n{w}_{1_i}}{n}$ 与 $\frac{{\sum }_{i=1}^m{w}_{2_i}}{m}$，小的先选。我们将 $w_i$ 的平均值称为 $c_i$。

考虑如何绑定节点的问题，一开始，每个序列只有一个点，它们的 $c_i$ 就是 $w_i$，每次操作把 $c_i$ 最大的点合并到他的父节点并记录贡献，一直合并到根节点即可。可以用优先队列实现。

Code

#include <bits/stdc++.h>
using namespace std;
const int N = 1000 + 5;
struct node {
    int par, sz, val;
}a[N];
int n, rt, fa[N], vis[N];
priority_queue<pair<double, int> >q;
int find(int x) {
    if (fa[x] == x) return x;
    return fa[x] = find(fa[x]);
} 
int main() {
  	while (~scanf("%d%d",&n, &rt) && n && rt){
      	int ans = 0;
      	for (int i = 1; i <= n; i++) {
       	  	scanf("%d", &a[i].val);
       	  	a[i].sz = 1, ans += a[i].val;
       	  	if (i != rt) q.push(make_pair(a[i].val, i));
      	}
   	   	for (int i = 1; i < n; i++) {
   	       	int x, val; scanf("%d%d", &x, &val);
   	    	a[val].par = x;
  	    }
  	    for (int i = 1; i <= n; i++) fa[i] = i, vis[i] = 0;
  	    while (!q.empty()) {
         	int x = q.top().second; q.pop();
          	if (vis[x]) continue;
          	vis[x] = 1;
          	int par = find(a[x].par);
          	ans += a[par].sz * a[x].val;
          	a[par].val += a[x].val;
       		a[par].sz += a[x].sz;
      	    if (par != rt) q.push(make_pair(1.0 * a[par].val / a[par].sz, par));
      		fa[x] = par;
      	}
      	printf("%d\n", ans);
    }
    return 0;
}