二叉搜索树的第k大节点

给定一颗二叉搜索树,找出第k大结点

  这道题也没什么好说的,就是二叉搜索树的中序遍历,找到第k大的数返回就可以了。不过需要注意的是,《剑指offer》这本书上使用C写的,所以迭代的时候,就直接使用的是k的引用,而Java中没有这种语法,所以这里我用了一个全局变量,稍微修改了一下,总之效果是一样的。

import org.junit.Test;
class TreeNode{
    int val;
    TreeNode left;
    TreeNode right;
    public TreeNode(int val){
        this.val = val;
    }
}
public class Solution{
    int kk = 0;
    public TreeNode kthNode(TreeNode root, int k){
        if (root == null || k == 0) {
            return null;
        }
        kk = k;
        return kthNodeCore(root);
    }
    private TreeNode kthNodeCore(TreeNode root){
        TreeNode target = null;
        if (root.left != null)
            target = kthNodeCore(root.left);

        if (target == null) {
            if (kk == 1) {
                // 用这种方法找到目标值,直接赋值,而不是直接迭代退栈,这里和之前用过的那些方法都不同
                target =  root;
            }
            kk--;
        }
        if (target == null && root.right != null) {
            target = kthNodeCore(root.right);
        }
        // 从这里返回找到的目标值
        return target;
    }

    @Test
    public void Test(){
        TreeNode root = new TreeNode(5);
        root.left = new TreeNode(3);
        root.left.left = new TreeNode(2);
        root.left.right = new TreeNode(4);

        root.right = new TreeNode(7);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(8);

        TreeNode node = kthNode(root, 8);
        // 这里是根据《剑指offer》书中的写法,直接返回结点,所以如果k大于了结点数,此时返回的是一个null
        // 这里需要判断一下。
        if (node != null)
            System.out.println(node.val);
        else
            System.out.println(-1);

    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值