版权声明:随意转载哦......但还是请注明出处吧: https://blog.csdn.net/dreaming__ldx/article/details/85110003
传送门
树形
题。
题意简述:给出一棵边权为1的树,允许选任意一个点
为根,求
的最大值。
直接统计出子树的权值和转移就行了。
代码:
#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;
}