[剑指offer]-二叉搜索树第k小结点

最新推荐文章于 2021-04-04 12:03:43 发布

原创最新推荐文章于 2021-04-04 12:03:43 发布 · 408 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer #Java

剑指offer 专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找第K小节点的算法。通过递归中序遍历的方式，每遍历一个节点就减小k的值，当k等于0时，当前节点即为第K小的节点。文章详细解释了如何利用TreeNode[]和int[]数组引用进行递归，避免值传递的问题，并讨论了如何在找到目标节点后减少后续的递归次数。

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

题目描述

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

思路
递归中序遍历，每遍历一个结点–k,当k=0时，当前结点即为所求结点。

注意点

使用TreeNode[] res和int[] kk的数组引用代入递归函数中，若为TreeNode类型和int 类型则为值传递，不起作用；或者可以使用成员变量。
找到所求结点后如何减少后面的递归次数。

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null||k<=0){
            return null;
        }
        TreeNode[] res=new TreeNode[1];
        int[] kk=new int[1];
        kk[0]=k;
        doKthNode( pRoot,kk, res);
        return res[0];
    }
    public void doKthNode(TreeNode pRoot, int[] k,TreeNode[] res){
    	//此处阻止所求结点上层结点继续遍历右子树
        if(k[0]<1){
            return;
        }
        if(pRoot.left!=null){
            doKthNode( pRoot.left, k, res);
        }
        k[0]--;
        if(k[0]==0){
            res[0]=pRoot;
            //此处阻止所求结点继续遍历右子树
            return;
        }
        if(pRoot.right!=null){
            doKthNode( pRoot.right, k, res);
        }
    }
}