LeetCode——230 二叉搜索树中第K小的元素(JAVA)

给定一个二叉搜索树的根节点 root,和一个整数 k,请你设计一个算法查找其中第 k个最小元素(从 1 开始计数)。

示例 1:

在这里插入图片描述

输入:root = [3,1,4,null,2], k = 1
输出:1

示例 2:

在这里插入图片描述

输入:root = [5,3,6,2,4,null,null,1], k = 3
输出:3

提示:

  • 树中的节点数为 n
  • 1 <= k<= n<= 104
  • 0 <= Node.val<= 104

思路

思路很简单,因为BST的中序遍历是按序递增的,因此,要查找第k个最小元素,其实就是中序遍历结果中的第k个数。

所以,只要设置一个成员变量放在外面(我这里写的是nowK),初始化为0,然后中序遍历每次访问到根结点时,先将nowK++,再看看自加完之后nowK与给定的k是否相等,如果相等,则说明已经查找到了第k个最小元素,记录下它的值即可。

代码

public class Solution {
	public class TreeNode {
		int val;
		TreeNode left;
		TreeNode right;
		TreeNode() {}
		TreeNode(int val) { this.val = val; }
		TreeNode(int val, TreeNode left, TreeNode right) {
			this.val = val;
		    this.left = left;
		    this.right = right;
		}
	}
	int nowK = 0;
	int num = 0;
	public void inOrder(TreeNode root, int k) {
		if(root.left!=null) {
			inOrder(root.left, k);
		}
		nowK++;
		if(nowK==k) {
			num = root.val;
		}
		if(root.right!=null) {
			inOrder(root.right, k);
		}
	}
	public int kthSmallest(TreeNode root, int k) {
		inOrder(root, k);
		return num;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值