2020吉林省赛-Problem I. World Tree（贪心）

题意： 给你一棵树，树上每个点有两个权值a[i]和b[i]，每经过一个节点u可以获得a[u]*Σb[v]的贡献，其中v是还未访问过的节点。问以哪种方式遍历整棵树，获得的最大贡献。

思路：

首先dfs出每个节点作u为根节点的子树中a和b的总和，记为s1[u] 和s2[u].

对于当前访问的u节点来说，它的子节点必定是还未访问过的。

考虑访问子节点的顺序，我们用贪心的思想：

对于当前节点u的两个子节点v1，v2来说，如果我们先访问v1再访问v2，以u为根的子树（未访问节点只考虑v2中的，不考虑当前节点的子树中的）产生的贡献是s1[v1]*s2[v2];

同理先访问v2再访问v1，产生的贡献是s1[v2]*s2[v1]。

于是我们可以排序来确定访问节点的顺序。

感觉这题的贪法跟acwing中耍杂技的牛想法一样……自己写的时候却没去这样想。

#include<bits/stdc++.h>
using namespace std;
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define endl '\n'
typedef long long LL;
const int N=100010,M=2*N,mod=1e9+7;
typedef pair<LL,int>PLI;
int n;
int h[N],e[M],ne[M],idx;
LL a[N],b[N],s1[N],s2[N];
LL res,sum; //sum代表未访问节点的b之和
void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs_solve(int u,int fa)
{
	s1[u]=a[u],s2[u]=b[u];
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(j==fa) continue;
		dfs_solve(j,u);
		s1[u]+=s1[j],s2[u]+=s2[j];
	}
}
bool cmp(int x1,int x2)
{
	return s1[x1]*s2[x2]>s1[x2]*s2[x1];
}
void dfs(int u,int fa)
{
	sum-=b[u];
	res+=a[u]*sum;
	vector<int>v;
	for(int i=h[u];~i;i=ne[i])
	{
		int j=e[i];
		if(j==fa) continue;
		v.push_back(j);
	}
	sort(v.begin(),v.end(),cmp);
	for(int i=0;i<v.size();i++)
		dfs(v[i],u);
}
int main()
{
	ios;
	int T;
	cin>>T;
	while(T--)
	{
		memset(h,-1,sizeof h);
		memset(s1,0,sizeof s1);
		memset(s2,0,sizeof s2);
		idx=0;
		cin>>n;
		sum=0,res=0;
		for(int i=1;i<=n;i++)
			cin>>b[i];
		for(int i=1;i<=n;i++)
			cin>>a[i];
		for(int i=1;i<=n;i++)
			sum+=b[i];
		for(int i=1;i<n;i++)
		{
			int x,y;
			cin>>x>>y;
			add(x,y),add(y,x);
		}
		dfs_solve(1,-1);
		dfs(1,-1);
		cout<<res<<endl;
	}
	return 0;
}