代码随想录二叉树——二叉搜索树中的众数

题目

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

结点左子树中所含结点的值小于等于当前结点的值
结点右子树中所含结点的值大于等于当前结点的值
左子树和右子树都是二叉搜索树
例如:

给定 BST [1,null,2,2],
在这里插入图片描述

返回[2].

提示:如果众数超过1个,不需考虑输出顺序

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

思路

二叉搜索树—>中序遍历有序

遍历有序数组的元素出现频率,从头遍历一定是相邻两个元素两两对比,然后把频率最高的元素输出就可以了。(可以使用双指针法,pre指向前一个节点,cur指向当前节点)

找众数(最大频率的元素集合)有两种方法:

  1. 两次遍历:先遍历一遍数组,找出最大频率,然后再遍历一遍数组把出现频率为maxCount的元素放进集合中(因为众数可以有多个,所以是集合)
  2. 一次遍历直接初始化maxCount,如果count=maxCount,那么把这个元素加入集合中,如果新统计的count > maxCount,那么不仅更新maxCount,而且清空结果集。
//递归法,采用一次遍历找众数,中序遍历
class Solution {
	ArrayList<Integer> resList;
	int maxCount;
	int	count;
	TreeNode pre;

	public int[] findMode(TreeNode root){
		resList = new ArrayList<>();//初始化集合
		maxCount = 0;
		count = 0;
		pre = null;
		findMode1(root);//从根节点开始,找出众数集合resList
		int[] res = new int[resList.size()];//将集合存储到数组中去,进行遍历
		for(int i = 0;i < resList.size(); i++){
			res[i] = resList.get(i);
		}
		return res;//返回有序数组
	} 
	
	public void findMode1(TreeNode root){
		if(root == null){
			return;
		}
		findMode1(root.left);//左,为中序遍历代码模板
		
		int rootValue = root.val;//记录根节点的值
		//计数
		if(pre == null || rootValue != pre.val){//如果前驱节点为空即第一个节点,或者当前节点的值不等于上一个节点即相邻两个元素不相等
			count = 1;//频率为1
		}else {//当前节点与前一个节点元素值相等
			count++;//频率+1
		}
		//更新结果以及maxCount
		if(count > maxCount){
			resList.clear();
			resList.add(rootValue);
			maxCount = count;//更新最大频率
		}
		else if(count == maxCount){
			resList.add(rootValue);
		}
		pre = root;//指针后移
		
		findMode1(root.right);//右,为中序遍历代码模板
	}
}

//中序遍历代码模板
void searchBST(TreeNode root){
	if(root == null){
		return;
	}
	searchBST();//左
	(处理节点的操作)//中
	searchBST();//右
	return;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
二叉搜索树,也叫二叉查找树(BST,Binary Search Tree),是一种特殊的二叉树。相对于普通的二叉树,它有一个特点:所有左子树的节点都比根节点小,所有右子树的节点都比根节点大。因此,它可以快速地进行查找、插入、删除等操作。 具体来说,二叉搜索树的定义如下: - 节点的左子树所有节点的值都小于该节点的值。 - 节点的右子树所有节点的值都大于该节点的值。 - 左右子树也都是二叉搜索树。 如下图所示,就是一个二叉搜索树的例子: ``` 8 / \ 3 10 / \ \ 1 6 14 / \ / 4 7 13 ``` 在这个树,每个节点都满足左子树的节点值小于该节点的值,右子树的节点值大于该节点的值。比如,节点 3 的左子树是 1 和 6,右子树是 4 和 7,都满足要求。 二叉搜索树的主要操作包括查找、插入和删除。 查找操作可以通过递归或者循环实现。递归实现如下: ``` def search(root, val): if not root or root.val == val: return root if root.val > val: return search(root.left, val) else: return search(root.right, val) ``` 插入操作需要先查找到插入的位置,然后创建一个新节点插入到该位置: ``` def insert(root, val): if not root: return TreeNode(val) if root.val > val: root.left = insert(root.left, val) else: root.right = insert(root.right, val) return root ``` 删除操作比较复杂,需要考虑多种情况。如果要删除的节点只有一个子节点,直接将其子节点替换上来即可。如果要删除的节点有两个子节点,可以找到其右子树的最小节点(或者左子树的最大节点)来替换该节点,然后再删除该最小节点(或者最大节点)。 ``` def delete(root, val): if not root: return None if root.val == val: if not root.left: return root.right elif not root.right: return root.left else: # 找到右子树的最小节点 p = root.right while p.left: p = p.left root.val = p.val root.right = delete(root.right, p.val) elif root.val > val: root.left = delete(root.left, val) else: root.right = delete(root.right, val) return root ``` 需要注意的是,在删除节点时,要保证删除后的树仍然是二叉搜索树

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

HDU-五七小卡

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值