poj2342 Anniversary party (树形dp，树的最大独立集)

问题描述：

Ural大学有N名职员，编号为1~N。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。
现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式

多组案例
第一行一个整数N。
接下来N行，第 i 行表示 i 号职员的快乐指数Hi。
接下来N-1行，每行输入一对整数L, K,表示K是L的直接上司。
最后一行输入0,0。

输出格式

输出最大的快乐指数。

数据范围

1≤N≤6000,
−128≤Hi≤127

Sample Input

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

Sample Output

5

分析:

最大独立集，只有选和不选两种情况
d(x)(1)表示选x可以得到的最大值
d(x)(0)表示不选x可以得到的最大值
找到根节点，从根开始dfs，回溯的时候更新d数组

code：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
const int maxm=6e3+5;
vector<int>g[maxm];
int in[maxm];
int d[maxm][2];
int n;
void init(){
    for(int i=1;i<=n;i++){
        in[i]=0;
        g[i].clear();
    }
}
void dfs(int x){
    for(int i=0;i<(int)g[x].size();i++){
        int v=g[x][i];
        dfs(v);
        d[x][0]+=max(d[v][0],d[v][1]);
        d[x][1]+=d[v][0];
    }
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    while(cin>>n){
        init();
        for(int i=1;i<=n;i++){
            cin>>d[i][1];
            d[i][0]=0;
        }
        while(1){
            int x,y;
            cin>>x>>y;
            if(x==0&&y==0)break;
            g[y].push_back(x);
            in[x]++;
        }
        for(int i=1;i<=n;i++){
            if(!in[i]){
                dfs(i);
                cout<<max(d[i][0],d[i][1])<<endl;//输出最大值
                break;
            }
        }
    }
    return 0;
}