算法导论 15-4 数形DP poj 2342

这个道题跟poj  2342上的一样。

题目大意：就是一个树，树的每个结点有个值，选了某个结点，就不能选择它的父结点，求整棵树选的节点值最大是多少。

解题思路：简单的树形DP

dp[i][0]表示不选i结点时，i子树的最大价值

dp[i][1]表示选i结点时，i子树的最大价值

那么列出状态方程

dp[i][0] = sum(max(dp[u][0], dp[u][1])) u 为结点i的儿子

dp[i][1] = sum(dp[u][0]) + val[i] ;    val[i]表示i结点的价值

 

最后求max(dp[root][0], dp[root][1])即为结果

 

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

const int maxn = 6005;

struct node
{
    int u, v, next;
};

node edge[maxn];
int n, e, head[maxn], val[maxn], indegree[maxn];
int dp[maxn][2];

void addEdge(int u, int v);
void dfs(int u);

int main()
{
   
    scanf("%d\n", &n);
    e = 0;
    memset(head, -1, sizeof(head));
    memset(indegree, 0, sizeof(indegree));
    memset(dp, 0, sizeof(dp));
    for(int i = 1; i <= n; i++)
        scanf("%d", &val[i]);
    int u, v;
    while(true)
    {
        scanf("%d %d", &v, &u);
        if(u == 0 && v == 0)
            break;
        addEdge(u, v);
        indegree[v]++;
    }
    int root;
    for(int i = 1; i <= n; i++)
    {
        if(indegree[i] == 0)
        {
            root = i;
            break;
        }
    }
    dfs(root);
    printf("%d\n", max(dp[root][0], dp[root][1]));
        return 0;
}

void addEdge(int u, int v)
{
    edge[e].u = u;
    edge[e].v = v;
    edge[e].next = head[u];
    head[u] = e++;
}

void dfs(int u)
{
    if(head[u] == -1)
    {
        dp[u][0] = 0;
        dp[u][1] = val[u];
        return;
    }
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].v;
        dfs(v);
        dp[u][0] += max(dp[v][0], dp[v][1]);
        dp[u][1] += dp[v][0];
    }
    dp[u][1] += val[u];
}