Kth Smallest Element in a BST

最新推荐文章于 2020-11-28 10:24:48 发布

cfcf0517

最新推荐文章于 2020-11-28 10:24:48 发布

阅读量327

点赞数

CC 4.0 BY-SA版权

分类专栏： java LeetCode

本文链接：https://blog.youkuaiyun.com/cfcf0517/article/details/47803363

java 同时被 2 个专栏收录

237 篇文章

订阅专栏

LeetCode

205 篇文章

订阅专栏

本文介绍了一种方法来找到给定二叉搜索树中的第k小元素，通过使用栈来遍历树并减小k的值直至找到目标元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a binary search tree, write a function kthSmallest to find the kth smallest

element in it.

Note:

You may assume k is always valid, 1 ≤ k ≤ BST's total elements.

找到二叉搜索树中的第k小的元素。

使用栈，如果有左子树，入栈，否则出栈，k--，k=0则当地节点就是要找的节点，

否则寻找右子树。

因为最小的点在最左下，所以一直操作左子树。

public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        Stack<TreeNode> stack=new Stack<TreeNode>();
		TreeNode p=root;
		while(p!=null||!stack.isEmpty()){
			if(p!=null){
				stack.push(p);
				p=p.left;
			}else{
				TreeNode t=stack.pop();
				k--;
				if(k==0)
					return t.val;
				p=t.right;
			}
		}
		return 0;
    }
}