[POJ2054]Color a Tree（贪心）

传送门
题意：有一棵树，每个节点都有一个权值Ci。现在要给每个点染色，第一个染根节点，其余的节点染色的时候其父节点必须已染色。每个节点染色会用掉一个时间单位，每个节点染色的代价是已经花的时间t*Ci。问染完全部节点所需要的最小代价。

---

跟着HNOI2018D2T2到的这里，发现是元问题。
真没想到是贪心。。不过是一个经典的贪心模型。
可以看这篇题解http://www.cnblogs.com/yu-chao/archive/2012/02/19/2358565.html
大体的贪心思路就是：按照类似平均数的$\frac{\sum{Ci}}{size}$（其中size是集合的大小），从大到小加入加入集合（因为大的肯定尽量在前面）即可。

---

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1010;
const int INF=1e9;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-') f=-1; ch=getchar();}
    while(ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
int num[N],sum[N];
int fa[N]; int findfa(int x){return (fa[x]!=x)?findfa(fa[x]):fa[x];}
int n,rt;
struct node{int x,y,next;}a[N*4]; int len,last[N];
void ins(int x,int y){a[++len].x=x; a[len].y=y;a[len].next=last[x];last[x]=len;}
bool vis[N];
int findmax()
{
    double maxx=0; int y=0;
    for(int i=1;i<=n;i++)
    {
        if(!vis[i] && maxx<double(sum[i])/num[i] && i!=rt)
        {
            maxx=double(sum[i])/num[i];
            y=i;
        }
    }
    vis[y]=true;
    return y;
}
void merge(int x,int fx)
{
    sum[fx]+=sum[x],num[fx]+=num[x];
    for(int k=last[x];k;k=a[k].next)
        fa[a[k].y]=fx;
}
int main()
{
    while(scanf("%d%d",&n,&rt)&& rt+n)
    {
        for(int i=1;i<=n;i++)
        {
            sum[i]=read();
            num[i]=1;
        }
        len=0; memset(last,0,sizeof(last));
        memset(vis,false,sizeof(vis));
        memset(fa,0,sizeof(fa));
        for(int i=1;i<n;i++)
        {
            int u=read(),v=read();
            ins(u,v);
            fa[v]=u;
        }

        int ans=0;
        for(int i=1;i<n;i++)
        {
            int x=findmax();//查询平均最大值 
            int fx=fa[x];
            while(vis[fx]) fx=fa[fx];//查询所在集合
            ans+=sum[x]*num[fx];
            merge(x,fx); 
        }
        ans+=sum[rt];
        printf("%d\n",ans);
    }
    return 0;
}