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

没有上司的晚会 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;   
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Eden_CY/article/details/78167366
文章标签: 树形dp
个人分类: DP ——树形DP
想对作者说点什么? 我来说一句

气氛值算法

2013年07月01日 206KB 下载

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭