给树上色（贪心与树）

每个节点都有一个“着色成本因子”，Ci。每个节点的着色成本取决于 Ci 和 Bob 完成该节点着色的时间。开始时，时间设置为0。如果节点i的着色完成时间为Fi，则节点i的着色成本为Ci * Fi。

例如，具有五个节点的树如图 1 所示。每个节点的着色成本因子为 1、2、1、2 和 4。Bob 可以按照 1、3、5、2、4 的顺序为树着色，最小总着色成本为 33。

给定一棵树和每个节点的着色成本因子，请帮助 Bob 找到为所有节点着色的最小可能的总着色成本。
输入
输入由几个测试用例组成。每个案例的第一行包含两个整数 N 和 R (1 <= N <= 1000, 1 <= R <= N)，其中 N 是树中的节点数，R 是根节点的节点数.第二行包含N个整数，其中第i个为Ci (1 <= Ci <= 500)，即节点i的着色成本因子。接下来的 N-1 行中的每一行都包含两个以空格分隔的节点号 V1 和 V2，它们是树中一条边的端点，表示 V1 是 V2 的父节点。不会列出两次边，将列出所有边。

N = 0 和 R = 0 的测试用例表示输入结束，不应进行处理。
输出
对于每个测试用例，输出一行包含 Bob 为所有节点着色所需的最小总着色成本。

思路：类似并查集把每次找到权值最大的节点然后向上合并到父亲节点上，子节点为0，父节点为子节点加上父节点除以num

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<set>
#include<string.h>
#include<vector>
using namespace std;
typedef long long ll;
const int N=1025;
struct node{
   int c,fa,num;
   double w;	
}p[N];
int n,r;
int find()
{
	double maxx=0;
	int res;
	for(int i=1;i<=n;i++){
		if(i!=r&&maxx<p[i].w){
			maxx=p[i].w;
			res=i;
		}
		
	}
	return res;
}
int main(){
	while(cin>>n>>r){
		ll ans=0;
		if(r==0&&n==0)break;
		for(int i=1;i<=n;i++){
			scanf("%d",&p[i].c);
			p[i].w=p[i].c;
			ans+=p[i].w;
			p[i].num=1;
		}
		int a,b;
		for(int i=1;i<n;i++){
			cin>>a>>b;
			p[b].fa=a;
		}
		for(int i=1;i<n;i++){
			int pos=find();
			int pa=p[pos].fa;
			ans+=p[pa].num*p[pos].c;
			for(int j=1;j<=n;j++){
				if(p[j].fa==pos)p[j].fa=pa;
			}
			p[pa].c+=p[pos].c;
			p[pa].num+=p[pos].num;
			p[pa].w=(double)1.0*p[pa].c/p[pa].num;
			p[pos].w=0;
		}
	printf("%lld\n",ans);
	}
	return 0;
}