给定一棵二叉搜索树,请找出其中第k大的节点。

会遍历,知道二叉搜索树特性就能做。

class Solution {
    List<Integer> list=new ArrayList<>();
    public int kthLargest(TreeNode root, int k) {
        rec(root);
        return list.get(list.size()-k);
    }
    public void rec(TreeNode root){
        if(root.left!=null){
            rec(root.left);
        }
        if(root==null){
           return;
        }
        list.add(root.val);
        if(root.right!=null){
            rec(root.right);
        }
    }
}

第一回刷没想着优化,
这里主要记录一下优化。

  1. 遍历到第K大,就应该停止了,把结果拿出来。
  2. 用到了额外的数组空间,能不能不用

注意是第K大,所以顺序右根左;第K小是左根右

class Solution {
    private int res=0,count=0;
    public int kthLargest(TreeNode root, int k) {
        rec(root,k);
        return res;
    }
    public void rec(TreeNode root,int k){
        if(root.right!=null){
            rec(root.right,k);
        }
        if(++count==k){
           
            res= root.val;
           return;
        }
    
         if(root.left!=null){
            rec(root.left,k);
        }
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值