P1352 没有上司的舞会

题目:P1352 没有上司的舞会
算法标签:dp,搜索,树形结构,记忆化搜索

从树的头往下求结果会有后效性,且有多个叶子节点,数据不易处理
则采用,叶子节点往头部求结果

对于某一节点,有选择和不选择两种情况
1、不选择的话 记 dp[ ][ 0 ]

for(int i = 0; i < a[x].tail.size(); i++){
		f[x][0] += max(f[son][0], f[son][1]);
	}

2、选择的话 记dp[ ][ 1 ]

for(int i = 0; i < a[x].tail.size(); i++){
		f[x][1] += f[son][0];
	}
#include <bits/stdc++.h>
using namespace std;
const int maxn = 6e3 + 5;
struct node {
	int r;
	vector<int> tail;
}a[maxn];
int n, head;
int f[maxn][3];
bool vis[maxn];
void dp(int x){
	f[x][0] = 0;
	f[x][1] = a[x].r;
	for(int i = 0; i < a[x].tail.size(); i++){
		int son = a[x].tail[i];
		dp(son);
		f[x][0] += max(f[son][0], f[son][1]);
		f[x][1] += f[son][0];
	}
}
int main(){
	cin >> n;
	int k, l, ans = 0;
	for(int i = 1; i <= n; i++){
		cin >> a[i].r;
	}
	for(int i = 1; i < n; i++){
		cin >> l >> k;
		a[k].tail.push_back(l);
		vis[l] = true;		// 标记该节点并非根
	}
	// 找出根
	for(int i = 1; i <= n; i++){
		if(!vis[i]){
			head = i;
			break;
		}
	}
	dp(head);
	cout << max(f[head][0], f[head][1]) << endl;
	return 0;
} ```

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值