树上DP - 洛谷P1352 - 没有上司的舞会

https://www.luogu.com.cn/problem/P1352

 

 

开二维数组  dp[ Maxsize ][ 2 ]

dp[ i ][ 0 ] 表示上司 i 不参加舞会的最优解， dp[ i ][ 1 ]不参加舞会的最优解。

易得方程：dp[ i ][ 0 ] = 每一个孩子节点 x :  dp[ i ][ 0 ]  += max( dp[ x ][ 0 ] , dp[ x ][ 1 ])  ;  dp[ i ][ 1 ] = 每一个孩子节点 x : dp[ i ][ 1 ]  += max( 0, dp[ x ][ 0 ])

记得在初始化时对每一个 dp[ i ][ 1 ] 初始化为其本身的点权

由于我们可以用dp[ i ][ 1 ]记录点权（同时也是初始化的过程）， 就不需要另外再开一个数组记录点权了，节省了空间。

#include <iostream>
#include <vector>
#define MAX(a,b) (a>b?a:b)
#define Maxsize 6000+1
using namespace std;
bool not_root[Maxsize];
bool vis[Maxsize];
int dp[Maxsize][2]; // dp[i][0] 不选用 i 的最优解 , dp[i][1] 选用 i 的最优解
vector<int> vec[Maxsize];
void dfs(int root){
    for (auto it = vec[root].begin(); it != vec[root].end(); it++) {
        if(!vis[*it]){
            vis[*it] = true;
            dfs(*it);
            vis[*it] = false;
        }
    }
    for (auto it = vec[root].begin(); it != vec[root].end(); it++) {
        if(vis[*it])continue;
        dp[root][0] += MAX(dp[*it][0],dp[*it][1]);
        dp[root][1] += MAX(dp[*it][0],0);
    }
}
int main(){
    int n;
    int a,b;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> dp[i][1];
    }
    for (int i = 1; i <= n; i++) {
        cin >> a >> b;  // 子 , 父
        vec[b].push_back(a);
        not_root[a] = true;
    }
    int root = -1;
    for (int i = 1; i <= n; i++) {
        if(!not_root[i]){
            root = i;
            break;
        }
    }
    dfs(root);
    cout << MAX(dp[root][0],dp[root][1]);
    return 0;
}