HDU1520Anniversary party(入门树形DP)

3 篇文章 0 订阅

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

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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值