poj2342树形dp

来源：互联网发布：Linux指令批量执行编辑：程序博客网时间：2024/04/29 22:04

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

思路：上下级关系是一棵树，设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 100000000vector<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;}

0 0