树形dp

为了庆祝乌拉尔州立大学建校80周年,将举行一个晚会。这所大学有雇员的等级结构。这意味着主管关系形成了一棵树,根植于校长V.E.Tretyakov。为了让每个人都觉得聚会有趣,校长不希望员工和他或她的直接上司都在场。人事处对每个员工的好客性进行了评估,所以每个人都有自己的号码(等级)。你的任务是列出一份客人的名单,尽可能多地列出客人的欢乐程度。

输入

雇员的编号从1到N。输入的第一行包含一个数字N.1<=N<=6000。随后的N行中的每一行都包含相应员工的欢乐度评级。欢乐等级是一个从-128到127的整数。之后是N–1行,用于描述主管关系树。树规范的每一行都有以下形式:

好的

这意味着第K个员工是第L个员工的直接主管。输入以行结束

0 0

输出

输出应包含客人评分的最大总和。
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

5

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<queue>
#include<stack>
#include<string.h>
using namespace std;
vector<int>G[10000];
const int p=100000;
int dp[p][2];
int fa[p];
void f(int u)
{
    for(int i=0;i<G[u].size();i++)
    {
    	int v=G[u][i];             //遍历所有人 
    	f(v);
	}
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		dp[u][0] = dp[u][0] + max(dp[v][1],dp[v][0]);   //进行选择判断是否参加 
	 	dp[u][1] = dp[u][1] + dp[v][0];
	}
} 
int main()
{
	int t; 
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		scanf("%d",&dp[i][1]);
		dp[i][0]=0;
		fa[i]=-1;
	}
	int u,v;  //u 员工  v老板 
	while(cin>>u>>v)
	{
		if(u == 0&& v == 0)
			break;
		G[v].push_back(u);
		fa[u] = v;
	} 
	int root=1;
	while(fa[root]!=-1) 
	root=fa[root];             //找到最终老板
	f(root);
	printf("%d\n",max(dp[root][1],dp[root][0]));
	return 0;
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
树形动态规划(Tree DP)是一种常用的动态规划算法,用于解决树结构相关的问题。在Python中,可以使用递归或者迭代的方式实现树形DP树形DP的基本思想是,从树的叶子节点开始,逐层向上计算每个节点的状态,并利用已经计算过的节点状态来更新当前节点的状态。这样可以通过自底向上的方式,逐步计算出整个树的最优解。 下面是一个简单的示例,演示如何使用树形DP解决一个二叉树中节点权值之和的最大值问题: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def max_sum(root): if root is None: return 0 # 递归计算左右子树的最大权值和 left_sum = max_sum(root.left) right_sum = max_sum(root.right) # 当前节点的最大权值和为当前节点值加上左右子树中较大的权值和 return root.val + max(left_sum, right_sum) # 构建一个二叉树 root = TreeNode(1) root.left = TreeNode(2) root.right = TreeNode(3) root.left.left = TreeNode(4) root.left.right = TreeNode(5) # 计算二叉树中节点权值之和的最大值 result = max_sum(root) print(result) ``` 这段代码中,我们定义了一个`TreeNode`类来表示二叉树的节点,其中`val`表示节点的权值,`left`和`right`分别表示左子节点和右子节点。`max_sum`函数使用递归的方式计算二叉树中节点权值之和的最大值,通过比较左右子树的最大权值和来确定当前节点的最大权值和。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值