洛谷 P1352 没有上司的舞会

题目传送门

解题思路:

用f[i][0]表示第i个人没去其本身与所有下属(包括间接)所能获得的最大值,f[i][1]表示第i个人去了其本身与所有下属(包括间接)所能获得的最小值,如果第i个人去了,则它的下属一定没去,如果第一个人没去,则它的下属去或不去皆可.

AC代码:

#include<iostream>
#include<cstdio>
#include<vector>

using namespace std;

int n,v[6001],f[6001][2];
vector<int> a[6001];
bool vis[6001];

inline void dfs(int root) {
    f[root][1] = v[root];
    if(a[root].empty()) return ;
    for(int i = 0;i < a[root].size(); i++) {
        dfs(a[root][i]);
        f[root][1] += f[a[root][i]][0];
        f[root][0] += max(f[a[root][i]][1],f[a[root][i]][0]); 
    }
}

int main() {
    scanf("%d",&n);
    for(int i = 1;i <= n; i++)
        scanf("%d",&v[i]);
    for(int i = 1;i < n; i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        a[y].push_back(x);
        vis[x] = 1;
    }
    for(int i = 1;i <= n; i++)
        if(!vis[i]) {
            dfs(i);
            printf("%d",max(f[i][0],f[i][1]));
        }
    return 0;
}