(蓝桥真题)左孩子右兄弟(树形DP)

该文通过贪心策略解决二叉树问题,利用左孩子兄弟表示法,计算以节点i为根的左孩子右兄弟二叉树的最大高度。通过深度优先搜索(DFS),遍历所有子节点,更新每个节点的最大深度,并返回最终答案。
摘要由CSDN通过智能技术生成

 样例输入:

5
1
1
1
2

样例输出:

4

分析:根据左孩子兄弟表示法我们可以发现对于一个节点而言,他的所有孩子节点只有一个作为左孩子,其余所有的孩子节点都作为右孩子往下建树,那么根据贪心的思想肯定会选择一个可能深度最小的孩子节点作为左子树节点,而选一个可能深度最大的节点作为右子树最后的一个节点,那么我们就可以令f[i]记录以i为根的左孩子右兄弟的二叉树的最大高度,那么我们每次向下搜索,都选择一个可能构造深度最大的孩子节点来作为右孩子,假如当前节点有t个孩子节点,那么其中一个孩子节点作为左子树节点,其余t-1个都是作为右子树节点,深度直接增加t,而且在此基础上加上最大可能孩子节点的深度即可完成当前节点的更新,最后直接返回答案即可。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;
const int N=2e5+10;
int f[N];//f[i]记录以i为根的左孩子右兄弟的二叉树的最大高度
vector<int> son[N];
int dfs(int x)
{
	if(x!=1) f[x]=1;
	for(int i=0;i<son[x].size();i++)
	{
		dfs(son[x][i]);
		f[x]=max(f[x],f[son[x][i]]+(int)son[x].size());
	}
	return f[x];
}
int main()
{
	int n;
	cin>>n;
	for(int i=2;i<=n;i++)
	{
		int t;
		scanf("%d",&t);
		son[t].push_back(i);
	}
	printf("%d",dfs(1)-1);
	return 0;
} 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
树上选点是蓝桥杯Java题目中的一种类型,通常需要在给定的树结构中选择一个或多个节点作为目标节点,并进行相应的操作。下面是一个简单的树上选点蓝桥Java题解的示例: 题目描述: 给定一棵有N个节点的树,每个节点上都有一个非负整数值。现在需要选择一些节点,使得选择的节点的值之和最大,且所选节点不能相邻(即选了一个节点,则其父节点和子节点都不能选)。请编写一个程序,计算出最大的节点值之和。 解题思路: 这是一个典型的动态规划问题。我们可以定义一个数组dp,其中dp[i]表示以第i个节点为根节点的子树中所选节点的最大值之和。对于每个节点i,有两种情况: 1. 选择节点i:则其子节点都不能选,所以dp[i] = val[i] + dp[grandchild1] + dp[grandchild2] + ... 2. 不选择节点i:则其子节点可以选择或不选择,所以dp[i] = max(dp[child1], dp[child2], ...) 根据以上思路,我们可以使用递归或者迭代的方式来计算dp数组。最终,所求的最大值即为dp,其中1表示根节点。 代码示例: ```java public class TreeSelectPoint { public static void main(String[] args) { int[] values = {0, 1, 2, 3, 4, 5}; // 节点值数组,下标从1开始 int[][] edges = {{1, 2}, {1, 3}, {2, 4}, {2, 5}}; // 树的边关系数组 int n = values.length - 1; // 节点个数 int[] dp = new int[n + 1]; // 动态规划数组 // 构建树的邻接表 List<List<Integer>> adjacencyList = new ArrayList<>(); for (int i = 0; i <= n; i++) { adjacencyList.add(new ArrayList<>()); } for (int[] edge : edges) { int u = edge[0]; int v = edge[1]; adjacencyList.get(u).add(v); adjacencyList.get(v).add(u); } dfs(1, -1, values, adjacencyList, dp); // 从根节点开始进行深度优先搜索 System.out.println(dp[1]); // 输出最大节点值之和 } private static void dfs(int cur, int parent, int[] values, List<List<Integer>> adjacencyList, int[] dp) { dp[cur] = values[cur]; // 初始化当前节点的dp值为节点值 for (int child : adjacencyList.get(cur)) { if (child != parent) { // 避免重复访问父节点 dfs(child, cur, values, adjacencyList, dp); dp[cur] += dp[child]; // 更新当前节点的dp值 } } } } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值