[poj2054]Color a Tree

对于权值最大的点，他一定会在父亲选完后立刻选，所以不断将其与父亲合并，直到只剩一个点就是答案
由于n比较少，可以直接暴力，如果n较大取最大值可以用堆，修改父亲可以用并查集

#include<cstdio>
#define N 1005
int n,r,x,y,ans,a[N],f[N],sz[N];
int main(){
    while (scanf("%d%d",&n,&r)!=EOF){
        if ((!n)&&(!r))return 0;
        ans=0;
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        for(int i=1;i<=n;i++)ans+=a[i];
        for(int i=1;i<=n;i++)sz[i]=1;
        for(int i=1;i<n;i++){
            scanf("%d%d",&x,&y);
            f[y]=x;
        }
        for(int i=1;i<n;i++){
            x=0;
            for(int j=1;j<=n;j++)
                if ((j!=r)&&((!x)||(a[x]*sz[j]<a[j]*sz[x])))x=j;
            y=f[x];
            ans+=sz[y]*a[x];
            for(int j=1;j<=n;j++)
                if (f[j]==x)f[j]=y;
            f[x]=y;
            a[y]+=a[x];
            a[x]=0;
            sz[y]+=sz[x];
        }
        printf("%d
",ans);
    }
}