P1352 没有上司的舞会 （树形dp

添加链接描述

#include<bits/stdc++.h>
using namespace std;
const int N=6e3+9;
int n,w[N];
int idx,e[N],ne[N],h[N],dp[N][2];
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u){
    dp[u][0]=0;
    dp[u][1]=w[u];
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        dfs(j);
        dp[u][0]+=max(dp[j][0],dp[j][1]);
        dp[u][1]+=dp[j][0];
    }
}
int vis[N];
int main(){ 
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add(b,a);
        vis[a]=1;
    }
    int root;
    for(int i=1;i<=n;i++){
        if(!vis[i])root=i;
    }
    // cout<<root<<endl;
    dfs(root);
    printf("%d",max(dp[root][0],dp[root][1]));
    return 0;
}

标签：idx,int,scanf,ne,vis,树形,P1352,dp
来源： https://blog.csdn.net/Minelois/article/details/120142095