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

最新推荐文章于 2022-07-01 09:13:53 发布

原创最新推荐文章于 2022-07-01 09:13:53 发布 · 226 阅读

0 ·

CC 4.0 BY-SA版权

刷题专栏收录该内容

55 篇文章

订阅专栏

该博客介绍了如何在二叉搜索树中找到第k大的节点。通过中序遍历，遵循右-中-左的顺序，可以有效地找到第k大的元素。在遍历过程中，一旦找到第k个最大元素，即可立即返回结果。提供的Java代码示例展示了这种解决方案。

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

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

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

这题应该也是中序遍历就OK了，但是这里由于是要拿第K大的元素，所以遍历顺序应该为右->中->左

class Solution {
    int res, k;

    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        dfs(root);
        return res;
    }

    void dfs(TreeNode root) {
        if (root == null) return;
        dfs(root.right);
        if (k == 0) return;  //提前返回
        if (--k == 0) res = root.val;
        dfs(root.left);
    }
}