poj2342树形dp

最新推荐文章于 2019-05-31 21:05:57 发布

天宇skyblue

最新推荐文章于 2019-05-31 21:05:57 发布

阅读量296

点赞数

分类专栏： dp

本文链接：https://blog.csdn.net/qq_24477135/article/details/50697100

版权

dp 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

题目大意：大学周年庆典聚会，每个人有一个欢乐度，每个人都不想和自己的上司碰面，问最大的欢乐度是多少。

思路：上下级关系是一棵树，设dp[i][0]表示不选i这个人，dp[i][1]选i这个人，那么显然dp[i][0] += max(dp[i->son][0] , dp[i->son][1]) , dp[i][1] += dp[i->son][0];

初始状态为dp[i][1]=res[i]，res[i]为每个人的欢乐度。

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>

using namespace std;
//#pragma comment(linker, "/STACK:102400000,102400000")
#define maxn 6005
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 100000000

vector<int>v[maxn];
int n;
int res[maxn];
int dp[maxn][2];
bool vis[maxn];

void solve(int root)
{
    vis[root] = 1;
    int up = v[root].size();
    if(up == 0) {dp[root][1] = res[root],dp[root][0] = 0;return;}
    for(int i = 0 ; i < up ; i ++)
    {
        if(!vis[v[root][i]])
        {
            solve(v[root][i]);
            dp[root][0] += max(dp[v[root][i]][0] , dp[v[root][i]][1]);
            dp[root][1] += dp[v[root][i]][0];
        }
    }
}

int main()
{
    while(scanf("%d" , &n)!= EOF)
    {
        for(int i = 0 ; i <= n ; i ++)v[i].clear();
        mem(dp, 0);mem(vis , 0);
        for(int i = 1 ; i <= n ; i ++) scanf("%d" , &res[i]);
        int l , k;
        while(scanf("%d %d" , &l , &k) && (l || k))
        {
            v[k].push_back(l);
            vis[l] = 1;
        }
        int root;
        for(int i = 1 ; i<= n ; i ++)if(!vis[i]) {root = i ; break;}
        for(int i = 1 ; i <= n ; i ++) dp[i][1] = res[i];
        mem(vis , 0);
        solve(root);
        int ans = max(dp[root][0] , dp[root][1]);
        printf("%d\n" , ans);
    }
    return 0;
}