关闭

没有上司的晚会(树形DP)

标签: 树形dp
135人阅读 评论(0) 收藏 举报
分类:

没有上司的晚会 poj2342

思路

一道比较水的树形DP
由题意得,两个相邻的节点不能同时取,而且存在负数
dp定义很直接
dp[i][1] =>取
dp[i][0] =>不取
dp数组存最大值
可以从根节点dfs下去 收集0和1的信息
状态转移也很显然:
dp[x][0]+=MAX(dp[y][0],dp[y][1]);
dp[x][1]+=dp[y][0];

题解

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;

#define FOR(i,a,b) for(int i=(a),i##_end_=(b);i<=i##_end_;++i)
#define DOR(i,a,b) for(int i=(a),i##_end_=(b);i>=i##_end_;--i)
#define INF 0x3f3f3f3f
#define LL long long
#define M 200005

inline void chkmi(int &a,int b){if(a>b)a=b;}
inline void chkmx(int &a,int b){if(a<b)a=b;}
inline int MAX(int a,int b){return a>b?a:b;}
inline int MIN(int a,int b){return a<b?a:b;}

vector<int> es[M];
int val[M],dp[M][2];
int n,a,b,ans;
void dfs(int s,int fr){
    dp[s][1]=val[s];
    FOR(i,0,es[s].size()-1){
        int y=es[s][i];
        if(y==fr)continue;
        dfs(y,s);
        dp[s][0]+=MAX(dp[y][0],dp[y][1]);
        dp[s][1]+=dp[y][0];
    }
}
int main(){
    cin>>n;
    FOR(i,1,n) scanf("%d",&val[i]);
    FOR(i,2,n){
        scanf("%d%d",&a,&b);
        es[a].push_back(b); 
        es[b].push_back(a);
    }
    dfs(1,0);
    printf("%d\n",MAX(dp[1][0],dp[1][1]));
    return 0;   
}
1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:954次
    • 积分:154
    • 等级:
    • 排名:千里之外
    • 原创:0篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条
    文章存档