码迷,mamicode.com
首页 > 其他好文 > 详细

HDU1520Anniversary party(入门树形DP)

时间:2015-08-29 17:02:10      阅读:132      评论:0      收藏:0      [点我收藏+]

标签:hdu1520

题意:给出了,参加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;
}

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

HDU1520Anniversary party(入门树形DP)

标签:hdu1520

原文地址:http://blog.csdn.net/u013167299/article/details/48088191

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!