poj 2054 Color a Tree(贪心)

题目大意：

给你一棵树，你要给这棵树染色，染色的代价为时间乘上接点的代价。还有一个要求是这个节点被染色一定要父亲也被染色。

思路：

显然我们第一个染根节点，只有一条链也很简单。但是如果有很多儿子要选择哪一个呢。仔细思考我们发现染色可以想成是一个合并过程。我们设now[i]为合并到i的所有结点费用的平均值，cnt[i]为节点i合并的节点数。每次染色后染色费用更新，这样合并n-1就好了，顺序就选出来了。

程序：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define N 1005
using namespace std;

int n,r,cnt1;
int a[N],last[N],fa[N],next[N],cnt[N];
double now[N];
struct tree{int to,next;}e[N*2];

void add(int x,int y){
	e[++cnt1].to=y; e[cnt1].next=last[x]; last[x]=cnt1;
	e[++cnt1].to=x; e[cnt1].next=last[y]; last[y]=cnt1;
}

void dfs(int x){
	for (int i=last[x];i;i=e[i].next){
		if (fa[e[i].to]==0){
			fa[e[i].to]=x;
			dfs(e[i].to);
		}
	}
}

void addedge(int x,int y){
	while (next[x]) x=next[x];
	next[x]=y;
}

void work(){
	memset(fa,0,sizeof(fa));
	fa[r]=-1;
	dfs(r);
	for (int i=1;i<=n;i++) now[i]=a[i];
	bool flag[N]; 
	int k,f;
	double max;
	memset(flag,1,sizeof(flag));
	memset(next,0,sizeof(next));
	for (int i=1;i<=n;i++) cnt[i]=1;
	for (int i=1;i<n;i++){
		max=0;
		for (int j=1;j<=n;j++)
		 if (j!=r&&flag[j]&&now[j]>max){
		 	max=now[j];
		 	k=j;
		 }
		int f=fa[k]; addedge(f,k);
		while (!flag[f]) f=fa[f];
		flag[k]=0;
		now[f]=(now[f]*cnt[f]+now[k]*cnt[k])/(cnt[k]+cnt[f]);
		cnt[f]+=cnt[k];
	}
	int p=r,ans=0;
	for (int i=1;i<=n;i++){
		ans+=i*a[p];
		p=next[p];
	}
	printf("%d\n",ans);
}

int main(){
	freopen("a.in","r",stdin);
	while (scanf("%d%d",&n,&r),n+r){
		for (int i=1;i<=n;i++) scanf("%d",&a[i]);
		memset(last,0,sizeof(last));
		cnt1=0;
		for (int i=1;i<n;i++) {
			int x,y;
			scanf("%d%d",&x,&y);
			add(x,y);
		}
		work();
	}
}