HDU1520(树型dp)

题意：

要开派对，邀请了上司就不能邀请他的下属，邀请了下属就不能邀请他的上司，每个人有一个值，求邀请的人的总值最大

思路：

树形dp:把每个人看成一个点，则该点有两个状态：邀请或没被邀请

定义dp[u][0]为节点没被邀请时的值；dp[u][1]为节点被邀请时的值

状态转移方程：

dp[u][0]=sum(max(dp[v][0],dp[v][1])//v为u的下属

dp[u][1]+=dp[v][0]//上司邀请了，则下属只有没被邀请这种状态

要求总值，则显然要求max(f[U][0],f[U][1])//U为根节点

用dfs遍历树用

代码：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <map>
#include <vector>
#include <string.h>
using namespace std;
const int N = 6005;
vector <int> G[N];
int value[N];
bool vis[N];
int dp[N][2];
int in[N];
void dfs(int root)
{
    vis[root] = 1;
    dp[root][1] = value[root];
    dp[root][0] = 0;
    for(int i = 0; i < (int)G[root].size(); i++)
    {
        int u = G[root][i];
        if(vis[u]) continue;
        dfs(u);
        dp[root][0]+=max(dp[u][0], dp[u][1]);
        dp[root][1]+=dp[u][0];

    }
    return;
}
int main()
{
    int n;
    while(~scanf("%d", &n) && n)
    {
        memset(vis, 0, sizeof(vis));
        memset(dp, 0, sizeof(dp));
        memset(in, 0, sizeof(in));
        for(int i=0;i<=n;i++)G[i].clear();
        for(int i = 1; i <= n; i++) scanf("%d", &value[i]);
        int v, u;
        while(~scanf("%d%d",&v,&u)&&(v+u))
        {
            G[u].push_back(v);
            in[v]++;
        }
        int root = 1;
        while(in[root]) {root++;}
        dfs(root);
        printf("%d\n",max(dp[root][0], dp[root][1]));
    }
    return 0;
}