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