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

版权声明：随意转载哦......但还是请注明出处吧： https://blog.csdn.net/dreaming__ldx/article/details/85110003

传送门
$sb$树形 $dp$题。
题意简述：给出一棵边权为1的树，允许选任意一个点 $v$为根，求 $\sum_{i=1}^ndist(i,v)*a_i$的最大值。

---

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

#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;
}