CF1092F Tree with Maximum Cost（dfs+dp)

果然我已经菜到被(div3)的题虐哭了

qwq

首先看到这个题，一个比较显然的想法就是先从1号点开始(dfs)一遍，然后通过一些奇怪的方式，再(dfs)一遍得到其他点的贡献。

那么具体应该这么做呢。
首先，我们维护两个数组(dis[i])表示(i)到1的距离，(sum[i])表示(i)的子树中的(val)的和。

然后我们考虑，如果从(fa[x])移动到(x)，相当于(x)的子树内的(dis)都要减一，对(ans)的贡献是(-sum[x])，然后(x)的子树外面的所有的点的(dis)要加一，对(ans)的贡献是(sum[1]-sum[x])。

那么我们只需要两遍(dfs)，第二遍(dfs)，一边(dfs)一边更新(ans)就好。

#include<bits/stdc++.h>
#define mk make_pair
#define pb push_back
#define ll long long
#define int long long

using namespace std;

inline int read()
{
  int x=0,f=1;char ch=getchar();
  while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
  while (isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
  return x*f;
}

const int maxn = 4e5+1e2;
const int maxm = 2*maxn;

int point[maxn],nxt[maxm],to[maxm];
int dis[maxn],sum[maxn];
int ans;
int n,m,cnt;
int tmp;
int val[maxn];

void addedge(int x,int y)
{
    nxt[++cnt]=point[x];
    to[cnt]=y;
    point[x]=cnt;
} 

void dfs(int x,int fa,int dep)
{
    sum[x]=val[x];
    dis[x]=dep;
    for (int i=point[x];i;i=nxt[i])
    {
        int p = to[i];
        if(p==fa) continue;
        dfs(p,x,dep+1);
        sum[x]+=sum[p];
    }
}

void solve(int x,int fa,int now)
{
    ans=max(ans,now);
    for (int i=point[x];i;i=nxt[i])
    {
        int p=to[i];
        if (p==fa) continue;
        solve(p,x,now+sum[1]-sum[p]-sum[p]); 
    }
}

signed main()
{
  n=read();
  for (int i=1;i<=n;i++) val[i]=read();
  for (int i=1;i<n;i++)
  {
  	int x=read(),y=read();
  	addedge(x,y);
  	addedge(y,x);
  }
  dfs(1,0,0);
  for (int i=1;i<=n;i++) tmp=tmp+dis[i]*val[i];
  ans=tmp;
  //cout<<ans<<endl;
  solve(1,0,tmp); 
  cout<<ans;
  return 0;
}