剑指 Offer 54. 二叉搜索树的第k大节点 题解

本文介绍了一种在二叉搜索树中寻找第K大元素的算法实现,通过中序遍历的方式将节点值有序排列,并提供两种方法:一种是利用栈存储遍历过程中的节点值再从中取出;另一种则是直接在遍历过程中计数并返回第K大的元素。

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

 

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
     Stack<Integer> stack1 = new Stack<>();
    public int kthLargest(TreeNode root, int k) {
        if(root==null){
            return -1;
        }
        //二叉搜索书是排好序的树 可以转换为中序遍历 就是从小到大排列
        //把遍历出的节点数值存储到栈中 拿到k就行了。
        midPrint(root);
        int temp=0;
        for(int i =0 ; i<k;i++){
            if(stack1.size()==0){
                return -1;
            }
            temp=stack1.pop();
        }
        return temp;
    }
    //递归的方法
    public void midPrint(TreeNode node){
        if(node.left!=null){
            midPrint(node.left);
        }
        stack1.push(node.val);
        if(node.right!=null){
            midPrint(node.right);
        }
    }
}

较慢 思考更高效率的方法:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
  
    public int kthLargest(TreeNode root, int k) {
         
        if(root==null){
            return -1;
        }
          Stack<TreeNode> stack1 = new Stack<>();
          TreeNode node=root;
          int count=0;
        //二叉搜索树是排好序的树 可以转换为中序遍历 就是从小到大排列
        //倒着遍历二叉树 遍历k个值。
            while(stack1!=null || node!=null){
                while(node!=null){
                    stack1.push(node);
                    node=node.right;
                }
                if(node==null){
                    node=stack1.pop();
                    if(++count==k){
                        return node.val;
                    }
                    node=node.left;
                }
            }
        
        return -1;
    }
   
}

效率一些的方法:使用中序输出的倒序,就是从大到小排列。也不用记录每一个数是多少。只需要设置一个count计数,等到count==k的时候获取那个值就可以了。 如果k输入混乱,比如k小于0或者大于树中节点数目,那么还需要加入判断,这个比较简单忘了写了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值