洛谷P1122 最大子树和

@Pelom  November 11, 2019

题意

求树上联通块的最大权值和

数据范围:$n \le 16000$

题解

树形$\text{dp}$
用$dp[u]$表示以$u$为根的子树中的最大权值和

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=16000+10;
int n,u,v;
int cnt,ans;
int head[N],a[N],dp[N];
struct edge{
    int nxt,to;
} e[N<<1];
inline void add(int u,int v){
    e[++cnt]=(edge){head[u],v};
    head[u]=cnt;
}
void dfs(int u,int f){
    dp[u]=a[u];
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v==f)
            continue;
        dfs(v,u);
        dp[u]+=max(0,dp[v]);
    }
    ans=max(ans,dp[u]);
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1,0);
    printf("%d",ans);
    return 0;
}

添加新评论