【LeetCode】剑指 Offer 54. 二叉搜索树的第k大节点

简介: 【LeetCode】剑指 Offer 54. 二叉搜索树的第k大节点

题目:

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

示例 1:

输入: root = [3,1,4,null,2], k = 1

  3

 / \

1   4

 \

  2

输出: 4

示例 2:


输入: root = [5,3,6,2,4,null,null,1], k = 3

      5

     / \

    3   6

   / \

  2   4

 /

1

输出: 4

限制:


1 ≤ k ≤ 二叉搜索树元素个数

分析:

          二叉搜索树的特点就是根左<根<根右,所以我们经过思考可知,通过中序遍历(左中右),得到的是二叉搜索树值的升序。


          因为我们要得到第K大的结点值,所以我们要值的降序排列,那么我们就用逆中序遍历(右中左),根据k值来记录当前是第几个结点。


代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int count,res;
 
    public int kthLargest(TreeNode root, int k) {
        this.count = k;
        method(root);
        return res;
 
    }
    public void method(TreeNode root) {
        if(root==null) return;
        if(count == 0) return;
        method(root.right);
        count--;
        if(count==0) {
            res = root.val;
        }
        method(root.left);
    }
}

因为进入递归之后记录数据k很乱,所以我们定义两个类变量count(用来记录当前是第几个结点)和res(用来存储第K大的值)。

思考:

之前我总是想不通为什么method(root.right);调用后要count-- 表示这个结点已经被遍历。


那method(root.left); 后面为什么不count-- 呢?


后来我想通了,我能提出这个问题说明我没懂递归的真谛。这个count--不是method(root.right);的count--。而是root的count-- 说明root这个结点被遍历到了。


递归整体可以这么理解,你就想先遍历一个结点(不带递归)

    public void method(TreeNode root) {
        if(root==null) return;
        if(count == 0) return;
 
        count--;
        if(count==0) {
            res = root.val;
        }
 
    }

当我把递归删掉后,我的目的就是遍历一个结点。

但是当我有遍历需求后,我想先看右边的,再遍历左边的。那么我就直接上下加个递归。

即:

    public void method(TreeNode root) {
        if(root==null) return;
        if(count == 0) return;
        method(root.right);
        count--;
        if(count==0) {
            res = root.val;
        }
        method(root.left);
    }

你可以将复杂的逻辑改成打印一个结点,那么我想先打印左再中再右。那么就上下加递归函数就可以了。

void method(TreeNode root) {
    if(root == null) return;
    method(root.left); //左
    System.out.println(root.val); //打印
    method(root.right); //右
}

就是这样。

目录
相关文章
|
2月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
45 6
|
2月前
|
算法
LeetCode第24题两两交换链表中的节点
这篇文章介绍了LeetCode第24题"两两交换链表中的节点"的解题方法,通过使用虚拟节点和前驱节点技巧,实现了链表中相邻节点的交换。
LeetCode第24题两两交换链表中的节点
|
2月前
|
算法
LeetCode第96题不同的二叉搜索树
文章介绍了LeetCode第96题"不同的二叉搜索树"的解法,利用动态规划的思想和递推公式,通过计算以任意节点为根的不同二叉搜索树的数量,解决了该问题。
LeetCode第96题不同的二叉搜索树
|
2月前
|
Python
【Leetcode刷题Python】剑指 Offer 30. 包含min函数的栈
本文提供了实现一个包含min函数的栈的Python代码,确保min、push和pop操作的时间复杂度为O(1)。
20 4
|
2月前
|
算法 Python
【Leetcode刷题Python】剑指 Offer 33. 二叉搜索树的后序遍历序列
本文提供了一种Python算法,用以判断给定整数数组是否为某二叉搜索树的后序遍历结果,通过识别根节点并递归验证左右子树的值是否满足二叉搜索树的性质。
17 3
|
2月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - II. 从上到下打印二叉树 II
本文提供了一种Python实现方法,用于层次遍历二叉树并按层打印结果,每层节点按从左到右的顺序排列,每层打印到一行。
30 3
|
2月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - I. 从上到下打印二叉树
本文介绍了使用Python实现从上到下打印二叉树的解决方案,采用层次遍历的方法,利用队列进行节点的访问。
30 2
|
6天前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
2月前
|
Python
【Leetcode刷题Python】剑指 Offer 26. 树的子结构
这篇文章提供了解决LeetCode上"剑指Offer 26. 树的子结构"问题的Python代码实现和解析,判断一棵树B是否是另一棵树A的子结构。
39 4
|
2月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
82 2