给树染色 贪心+树的结点合并

Posted 2021-10-17 karshey

题
y总讲解

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 

//
const int N=1e3+10;
int n,root;
struct node
{
	int father,val,sum;
	double avg;
}nodes[N];

int find()//找当前权值最大的 
{
	int res=0;
	double maxn=-1;
	for(int i=1;i<=n;i++)
	{
		if(maxn<nodes[i].avg)
		{
			maxn=nodes[i].avg;
			res=i;
		}
	}
	return res;
}

int main()
{
	cin>>n>>root;
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		auto &nd=nodes[i];
		cin>>nd.val;
		nd.sum=1;
		nd.avg=nd.val; 
		ans+=nd.val;
	}
	
	for(int i=1;i<=n-1;i++)
	{
		int a,b;cin>>a>>b;
		nodes[b].father=a;		
	}
	
	for(int i=1;i<=n;i++)
	{
		int now=find();//找到当前情况下最大权值的位置
		int f=nodes[now].father; 
		ans+=nodes[now].val*nodes[f].sum;
		nodes[now].avg=-1;//删除节点
		for(int j=1;j<=n;j++)
		{
			if(nodes[j].father==now)
			{
				nodes[j].father=f;
			}
		} 
		nodes[f].val+=nodes[now].val;
		nodes[f].sum+=nodes[now].sum;
		nodes[f].avg=(double)nodes[f].val/nodes[f].sum;
	}
	
	cout<<ans;
	return 0; 
}