LeetCode Java刷题笔记—剑指 Offer 54. 二叉搜索树的第k大节点

Posted 2022-08-06 刘Java

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

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

简单难度，我们使用反中序遍历，然后遍历一次进行k自减1，当k为0时对应的元素就是第k大的节点。

int k, res;

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


private void dfs(TreeNode root) 
    //右递归
    if (root.right != null && k > 0) 
        dfs(root.right);
    
    //中序递减k并判断是否为0
    if (--k == 0) 
        res = root.val;
        return;
    
    //左循环
    if (root.left != null && k > 0) 
        dfs(root.left);