hdu 1055 && poj 2054

/* hdu 1055&&poj 2054 给定一棵树上访问  但是要求是先访问了父亲节点才能访问子节点   首先是从根节点开始的
每个节点都有一个权值，访问的时间是递加的，访问的费用是每个节点的权值*访问的顺序，问最后最小花费*/
/*贪心的思想: 采用累积的想法，一开始的时候“权值”=权值，这样的话我们一定要先找到当前的“权值”的最大值 然后*他父亲的访问时间，因为
访问完父亲节点一定先访问它，然后要把他的所有的子节点都指向他的父亲节点， 这个访问之后“权值”设定为0  因为这个已经访问过了，因为这个时候我们并没有保证已经访问过他的父亲节点
（所以这里时间要加上   权值也要加上儿子的权值  然后“权值”取的是二者相除的结果）这里的操作是为了把前面这个落下的都加回去，这样一直到访问结束
*/
#include <iostream>
#include <cstdio>

using namespace std;

const int N = 1007;

struct node
{
    int t;    //time[]
    int p;    //记录父节点
    int c;    //c[]
    double w;    //w[]
}num[N];

int n, r;

int find()    //查找最大的权值，返回其位置
{
    int pos;
    double max = 0;
    for(int i = 1; i <= n; i++)
        if(num[i].w > max && i != r)    //pos不能为根节点
        {
            max = num[i].w;
            pos = i;
        }
    return pos;
}

int main()
{
    freopen("in.txt", "r", stdin);
    int i, j, a, b, pos, ans, f;
    while(scanf("%d%d", &n, &r), n||r)
    {
        ans = 0;
        for(i = 1; i <= n; i++)
        {
            scanf("%d", &num[i].c);
            num[i].w = num[i].c;    //初始时w[i]置为c[i];
            num[i].t = 1;        //time[i] 置为1；
            ans += num[i].c;    //初始ans = sum(c[i]);
        }
        for(i = 1; i < n; i++)
        {
            scanf("%d%d", &a, &b);
            num[b].p = a;    //记录父节点，建立树关系
        }
        i = n;
        while(i > 1)
        {
            pos = find();    //找到最大权值的位置
            num[pos].w = 0;        //找到后将之置0，否则影响下次查找。
            f = num[pos].p;        //f为父节点
            ans += num[pos].c * num[f].t;    //将找到的c[pos]*time[f]加入ans
            for(j = 1; j <= n; j++)
                if(num[j].p == pos)
                    num[j].p = f;    //如果有节点与pos相连，让它指向pos的父节点
            num[f].c += num[pos].c;        //C_i = c[该节点] + c[父节点]
            num[f].t += num[pos].t;        //T_i = time[该节点]+time[父节点]    该节点所占的比例
            num[f].w = (double)num[f].c/num[f].t;    //合并后的f节点的权值    该节点所占的权重  到时候*t直接就加回去了
            i--;
        }
        printf("%d\n", ans);
    }
    return 0;
}