Kth Smallest Element in a BST

最新推荐文章于 2022-04-14 14:33:13 发布

最新推荐文章于 2022-04-14 14:33:13 发布 · 107 阅读

leetcode 专栏收录该内容

193 篇文章

订阅专栏

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

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;
    }
}