hdu 1055 color on a tree



题意：给一棵树染色。对于一个节点，只有当父亲节点被染色后才能给它染色。给一个节点染色需要1s，花费为染色结束的时刻乘以该节点权值。初始时刻为0

求最小花费。

思路：贪心+并查集

有关证明及结论：http://blog.csdn.net/a601025382s/article/details/38780507

#include<cstdio>
using namespace std;
typedef double db;
struct P{
    int c,fa,t;
    db w;
}p[1005];

int main()
{
    int n,i,j,ans,r,x,y;
    while(~scanf("%d%d",&n,&r)&&(n+r))
    {
        ans=0;
        for(i=0;i<n;++i)
        {
            scanf("%d",&p[i].c);
            ans+=p[i].c;
            p[i].w=p[i].c;
            p[i].fa=i;
            p[i].t=1;
        }
        for(i=0;i<n-1;++i)
        {
            scanf("%d%d",&x,&y);
            p[y-1].fa=x-1;
        }
        for(i=0;i<n-1;++i)
        {
            db M=-1;
            int pos,fa;
            for(j=0;j<n;++j)
                if(p[j].w>M&&j!=r-1) M=p[j].w,pos=j;
            p[pos].w=0;
            fa=p[pos].fa;
            ans+=p[pos].c*p[fa].t;
            for(j=0;j<n;++j)
                if(p[j].fa==pos) p[j].fa=fa;
            p[fa].c+=p[pos].c;
            p[fa].t+=p[pos].t;
            p[fa].w=1.0*p[fa].c/p[fa].t;
        }
        printf("%d\n",ans);
    }
    return 0;
}