POJ 2342. Anniversary party 简单树形DP c++ 代码

最新推荐文章于 2025-07-28 16:57:20 发布

原创最新推荐文章于 2025-07-28 16:57:20 发布 · 500 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#POJ #树形DP #动态规划 #c++

数据结构与算法专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用树形动态规划解决最优选择问题的方法。通过递归地遍历树结构并应用动态规划，算法能够高效地找到针对每个节点的最大收益。具体实现包括初始化动态规划数组、递归构建树结构并进行状态转移等步骤。

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxN 6005

int N, L, K;
int dp[maxN + 1][2], father[maxN + 1];
bool visited[maxN];

void treeDp(int index) {
	if (visited[index]) return;
	visited[index] = true;
	for (int i = 1; i <= N; i++) {
		if (father[i] == index) {
			treeDp(i);
			dp[index][0] += max(dp[i][1], dp[i][0]);
			dp[index][1] += dp[i][0];
		}
	}
}

int main(void) {
	memset(dp, 0, sizeof(dp));
	memset(father, -1, sizeof(father)); 
	memset(visited, 0, sizeof(visited));

	cin >> N;
	for (int i = 1; i <= N; i++) {
		cin >> dp[i][1];
	}

	while (cin >> L >> K && !(L == 0 && K == 0)) 
		father[L] = K;

	// 找根节点
	int root = 1;
	while (father[root] != -1) 
		root = father[root];

	treeDp(root);
	
	cout << max(dp[root][0], dp[root][1]) << endl;
}