HDU 1520 Anniversary party （基础树形DP）

最新推荐文章于 2024-05-22 17:32:19 发布

原创最新推荐文章于 2024-05-22 17:32:19 发布 · 430 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树DP #DP #HDU

HDU 同时被 3 个专栏收录

7 篇文章

订阅专栏

2 篇文章

订阅专栏

树DP

1 篇文章

订阅专栏

本文详细解析了HDU 1520问题的解决方案，通过树形动态规划的方法来求解最大值问题。利用C++实现代码展示了如何构造树结构并进行深度优先搜索以计算最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1520

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

const int N = 6666;
int v[N], fa[N], dp[N][2];
vector<int> vec[N];

void dfs(int rt)
{
    int len = vec[rt].size();
    dp[rt][1] = v[rt];
    for(int i=0; i<len; i++) dfs(vec[rt][i]);
    for(int i=0; i<len; i++) {
        dp[rt][0] += max(dp[vec[rt][i]][0], dp[vec[rt][i]][1]);
        dp[rt][1] += dp[vec[rt][i]][0];
    }
}

int main()
{
    int n, a, b;
    while(~scanf("%d", &n)) {
        for(int i=1; i<=n; i++) {
            scanf("%d", &v[i]);
            vec[i].clear();
            fa[i] = -1;   // 初始化树根标记
            dp[i][0] = dp[i][1] = 0;
        }
        while(scanf("%d%d", &a, &b), a||b) {
            fa[a] = b;
            vec[b].push_back(a);  // 建立结点间的父子关系
        }
        int rt = 1;
        while(fa[rt] != -1) rt = fa[rt];  // 寻找树根
        dfs(rt);
        printf("%d\n", max(dp[rt][0], dp[rt][1]));
    }
    return 0;
}