Kth Smallest Element in a BST

iteye_6755

于 2015-10-31 20:46:21 发布

阅读量98

点赞数

分类专栏： leetcode

leetcode 专栏收录该内容

193 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找第K小元素的方法。通过使用栈来辅助实现中序遍历，可以有效地找到目标元素。此算法确保了遍历顺序为升序，便于找到第K小的节点。

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

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public int kthSmallest(TreeNode root, int k) {
        if (root != null) {
        	Stack<TreeNode> stack = new Stack<>();
        	while (!stack.isEmpty() || root != null) {
        		if (root != null) {
        			stack.push(root);
        			root = root.left;
        		} else {
        			root = stack.pop();
        			if (--k == 0) {
        				return root.val;
        			}
        			root = root.right;
        		}
        	}
        }
        return 0;
    }
}