2021-07-18

Leetcode暑假训练100题———第五题

剑指offer 54

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

题目分析：由于是二叉搜索树，所以使用中序遍历，即可将结点进行从小到大排序。创建一个list来按序存储结点的值，最后直接返回第k大的结点。
代码如下：

// JAVA
    List<Integer> list = new ArrayList<>();
    public int kthLargest(TreeNode root, int k) {
        dfs(root);
        // System.out.println(list);
        return list.get(list.size()-k);
    }

    public void dfs(TreeNode root){
        if(root == null) return;
        
        dfs(root.left);
        list.add(root.val);
        dfs(root.right);
    }

改进优化版本（空间复杂度降低，减少部分不必要的遍历）

// JAVA
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) 
            {
                res = root.val;
                return;
            }
        dfs(root.left);
    }
}