HDU1520Anniversary party（入门树形DP）

题意：给出了，参加party的人的关系，但是如果一个人的直接上司出席，那么他就不会出席，每个人有自己的rating。现在想要rating最大。

在网上找了半天，也没找到比较好的树形DP讲解，暂时的感觉是利用DFS搜索的时候，回溯更新。

1.dp[u][0]:表示u这个人不去，那么他的手下v可以选择去或者不去。所以 dp[u][0]+=max(dp[v][0],dp[v][1]);

2.dp[i][1]:表示u这个人去， 那么由题意他的手下肯定不会去，所以dp[u][1]+=dp[v][0];

状态表示，以及状态的转移方程有了，下面就是DFS这颗关系树了。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<string>
#include<cstring>
#include<stack>
#include<queue>
#include<vector>
#include<cstdlib>
#define lson (rt<<1),L,M
#define rson (rt<<1|1),M+1,R
#define M ((L+R)>>1)
#define cl(a,b) memset(a,b,sizeof(a));
#define LL long long
#define P pair<int,int>
#define X first
#define Y second
#define pb push_back
#define fread(zcc)  freopen(zcc,"r",stdin)
#define fwrite(zcc) freopen(zcc,"w",stdout)
using namespace std;
const int maxn=6005;
const int inf=999999;

int a[maxn];
int in[maxn];
vector<int> G[maxn];
int dp[maxn][2];
void Tree_dfs(int u){
    dp[u][0]=0;//u不去
    dp[u][1]=a[u];//u去的话，首先会有自己的rating
    int N=G[u].size();
    for(int i=0;i<N;i++){//遍历所有子节点
        int v=G[u][i];
        Tree_dfs(v);
        dp[u][0]+=max(dp[v][0],dp[v][1]);//回溯的时候，从叶子节点更新到父亲节点
        dp[u][1]+=dp[v][0];//也是回溯的更新
    }
}

int main(){
    int n;
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            G[i].clear();
            in[i]=0;
            dp[i][0]=dp[i][1]=0;
        }

        int x,y;
        while(scanf("%d%d",&x,&y)&&(x+y)){
            G[y].pb(x);//y是x的上司，不要写反了
            in[x]++;//顶点的入度
        }

        int root=1;
        for(int i=1;i<=n;i++)if(in[i]==0){//入度是0的点是根节点
            root=i;break;
        }

        Tree_dfs(root);//从根节点出发搜索
        printf("%d\n",max(dp[root][0],dp[root][1]));//答案
    }

    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。