剑指offer-62：二叉搜索树的第k个结点

最新推荐文章于 2022-03-05 16:24:47 发布

原创最新推荐文章于 2022-03-05 16:24:47 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

67 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找第K小节点的算法。通过使用栈进行中序遍历，可以有效地找到目标节点。示例代码展示了如何实现这一过程。

题目描述

给定一棵二叉搜索树，请找出其中的第k小的结点。例如，（5，3，7，2，4，6，8）中，按结点数值大小顺序第三小结点的值为4。

代码

public class Solution62 {


    TreeNode KthNode(TreeNode pRoot, int k) {


        if (pRoot == null || k <= 0)
            return null;
        int i = 1;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode p = pRoot;
        while (p != null || !stack.isEmpty()) {
            while (p != null) {
                stack.push(p);
                p = p.left;
            }
            p = stack.pop();
            if (i++ == k)
                return p;
            p = p.right;
        }
        return null;
    }

}