结点选择(树形DP)

在这里插入图片描述从任一根节点开始dfs,其中fy维护选择当前节点时对应的最大权值,fn维护不选择当前节点时对应的最大权值,最后返回fy[根]与fn[根]的最大值即可。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include<vector>
using namespace std;
using namespace std;

const int maxn = 100000 + 5;
int fy[maxn];  //表示选择当前节点对应的最大权值
int fn[maxn];  //表示不选择当前节点对应的最大权值
int vis[maxn];
vector<int>v[maxn];

void dfs(int u) {
	vis[u] = 1;
	int size = v[u].size();
	for (int i = 0; i < size; i++) {
		if (!vis[v[u][i]]) {
			dfs(v[u][i]);
			fy[u] += fn[v[u][i]];
			fn[u] += max(fn[v[u][i]], fy[v[u][i]]);
		}
	}
}


int main()
{
	int n; cin >> n;
	for (int i = 1; i <= n; i++)
		scanf("%d", fy + i);
	int v1, v2;
	for (int i = 1; i < n; i++) {
		scanf("%d%d", &v1,&v2);
		v[v1].push_back(v2);
		v[v2].push_back(v1);
	}
	dfs(v1);
	cout << max(fy[v1], fn[v1]);
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值