没有上司的舞会 / 树上求和【树形DP】

最新推荐文章于 2025-02-15 22:33:25 发布

原创最新推荐文章于 2025-02-15 22:33:25 发布 · 102 阅读

1 ·

CC 4.0 BY-SA版权

DP 同时被 2 个专栏收录

125 篇文章

订阅专栏

图论

50 篇文章

订阅专栏

本文介绍了如何解决树形结构中的求和问题，通过一道洛谷P1352题目来展示经典解题思路。代码中展示了利用深度优先搜索（DFS）进行节点信息传递和求解最大路径和的方法。

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

>Link

luogu P1352

ybtoj树上求和

>解题思路

经典例题划水

>代码

**#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 6010
using namespace std;

struct edge
{
	int to, nxt;
} e[N * 2];
int n, a[N], f[N][5], cnt, h[N];

void dfs (int now, int fath)
{
	for (int i = h[now]; i; i = e[i].nxt)
	{
		int v = e[i].to;
		if (v == fath) continue;
		dfs (v, now);
		f[now][1] += f[v][0];
		f[now][0] += max (f[v][1], f[v][0]);
	}
	f[now][1] += a[now];
}

int main()
{
	int x, y;
	scanf ("%d", &n);
	for (int i = 1; i <= n; i++) scanf ("%d", &a[i]);
	for (int i = 1; i < n; i++)
	{
		scanf ("%d%d", &x, &y);
		e[++cnt] = (edge){y, h[x]}; h[x] = cnt;
		e[++cnt] = (edge){x, h[y]}; h[y] = cnt;
	}
	dfs (1, 0);
	printf ("%d", max (f[1][1], f[1][0]));
	return 0;
}**