【*1900 换根DP】CF1092F

感觉很简单，根本没有1900的难度 

CF1092F Tree with Maximum Cost - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)y

题意：

思路：

考虑换根DP

首先先树形DP，然后dfs换根

设dp[u]为在以u为根的子树上，所有点的贡献

dp[u]=dp[v]+sz[v]

然后考虑换根，可以注意到u子树部分深度加1，v子树部分深度-1，因此方程就是dp[v]=dp[u]+sum-sz[v]-sz[v] 

如果要直观算贡献，一般是将u子树去掉v的那部分和v子树分开来考虑变化量

如果实在想不清楚，可以手推样例：（这里是样例1）

Code：

#include 

#define int long long

using namespace std;

using i64 = long long;

const int mxn=1e6+10;
const int mxe=1e6+10;
const int mod=1e9+7;

struct ty{
    int to,next;
}edge[mxe<<2];

int N,u,v;
int tot=0,sum=0;
int a[mxn];
int head[mxn];
int dp[mxn],sz[mxn];

void add(int u,int v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void G_init(){
    tot=0;
    for(int i=0;i<=N;i++){
        head[i]=-1;
    }
}
void dfs1(int u,int fa){
    sz[u]=a[u];
    for(int i=head[u];~i;i=edge[i].next){
        if(edge[i].to==fa) continue;
        dfs1(edge[i].to,u);
        sz[u]+=sz[edge[i].to];
        dp[u]+=dp[edge[i].to]+sz[edge[i].to];
    }
}
void dfs2(int u,int fa){
    for(int i=head[u];~i;i=edge[i].next){
        if(edge[i].to==fa) continue;
        dp[edge[i].to]=dp[u]-sz[edge[i].to]+sum-sz[edge[i].to];
        dfs2(edge[i].to,u);
    }
}
void solve(){
    cin>>N;
    G_init();
    for(int i=1;i<=N;i++) cin>>a[i],sum+=a[i];
    for(int i=1;i<=N-1;i++){
        cin>>u>>v;
        add(u,v);
        add(v,u);
    }
    dfs1(1,0);
    dfs2(1,0);
    int ans=0;
    for(int i=1;i<=N;i++) ans=max(ans,dp[i]);
    cout<>__;
    while(__--)solve();return 0;
}