2018.12.19 codeforces 1092F. Tree with Maximum Cost（换根dp）

传送门

sbsbsb树形dpdpdp题。

题意简述：给出一棵边权为1的树，允许选任意一个点vvv为根，求∑i=1ndist(i,v)∗ai\sum_{i=1}^ndist(i,v)*a_i∑i=1n​dist(i,v)∗ai​的最大值。

---

直接统计出子树的权值和转移就行了。

代码：

#include<bits/stdc++.h>
#define ri register int
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
const int N=2e5+5;
typedef long long ll;
int a[N],n;
vector<int>e[N];
ll sum[N],ans=0,sig=0;
inline void dfs1(int p,int fa,int dep){
	sum[p]=a[p],sig+=(ll)a[p]*dep;
	for(ri i=0;i<e[p].size();++i)if(e[p][i]^fa)dfs1(e[p][i],p,dep+1),sum[p]+=sum[e[p][i]];
}
inline void dfs2(int p,int fa,ll Sum){
	ans=max(ans,Sum);
	for(ri i=0;i<e[p].size();++i)if(e[p][i]^fa)dfs2(e[p][i],p,Sum-2ll*sum[e[p][i]]+sum[1]);
}
int main(){
	n=read();
	for(ri i=1;i<=n;++i)a[i]=read();
	for(ri i=1,u,v;i<n;++i)u=read(),v=read(),e[u].push_back(v),e[v].push_back(u);
	dfs1(1,0,0),dfs2(1,0,sig);
	cout<<ans;
	return 0;
}