POJ2054 Color a Tree 贪心

容易知道，除了根节点之外，下面的儿子节点如果是全树权值最大的点一定会在父亲节点染色之后被染色。

假如现在有x，y，z三个节点，现在知道x和y的染色是连续的。

那么先染x，y再染z的代价是x+2y+3z

如果先z后x和y那么代价是z+2x+3y

都加上(z-y)后再除以2，就是(x+y)/2+2z和z+(x+y)

也就是(x+y)/2和z谁先染色的代价

于是同理可知多个点的合并节点等价代价为sum/点数

然后不断合并就好了

/* ***********************************************
Author        :BPM136
Created Time  :2018/7/16 11:36:59
File Name     :2054.cpp
************************************************ */

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<vector>
using namespace std;

typedef long long ll;
typedef vector<int> VI;

const int N = 1005;

int c[N],a[N];
int fa[N];
int n,root;
VI b[N];
bool vis[N];

int main() {
    while(scanf("%d%d",&n,&root)!=EOF) {
        if(n==0 && root==0) return 0;
        for(int i=1;i<=n;i++) {
            scanf("%d",&c[i]);
            a[i]=c[i];
        }
        for(int i=1;i<n;i++) {
            int p,f;
            scanf("%d%d",&f,&p);
            fa[p]=f;
        }
        for(int i=1;i<=n;i++) {
            b[i].clear();
            b[i].push_back(i);
        }
        for(int i=1;i<=n;i++) vis[i]=0;
        vis[root]=1;
        for(int o=1;o<n;o++) {
            int v=0;
            double mx=0.0;
            for(int i=1;i<=n;i++) if(vis[i]==0) {
                if(a[i]*1.0/b[i].size()>mx) {
                    mx=a[i]*1.0/b[i].size();
                    v=i;
                }
            }
            int f=fa[v];
            vis[v]=1;
            for(int i=0;i<b[v].size();i++) b[f].push_back(b[v][i]);
            a[f]+=a[v];
            for(int i=1;i<=n;i++) if(vis[i]==0 && fa[i]==v) {
                fa[i]=f;
            }
        }
        ll ans=0;
        for(int i=0;i<b[root].size();i++) ans+=(long long)(i+1)*c[ b[root][i] ];
        printf("%lld
",ans);
    }
    return 0;
}