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

63 阅读 0 评论 42 点赞

我是靠谱客的博主英俊金鱼，最近开发中收集的这篇文章主要介绍剑指 Offer 54. 二叉搜索树的第k大节点，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

在这里插入图片描述
思路：二叉搜索树的中序遍历是递增序列，将中序遍历的结果保存下来，返回中序遍历的倒数第k个元素即可。
二叉搜索树第 k 大的节点可转化为求此树的中序遍历的倒数第 k 个节点

class Solution {    
    public int kthLargest(TreeNode root, int k) {
        List<Integer> res = new ArrayList<>();
        inorder(root, res);
        // 返回中序遍历的倒数第k个元素
        return res.get(res.size() - k);
    }

    public void inorder(TreeNode root, List<Integer> list){
        if (root == null)   return ;
        // 左
        inorder(root.left, list);
        // 根
        list.add(root.val);
        // 右
        inorder(root.right, list);
    }
}

改进：中序遍历是 左->根->右，如果我们右->根->左遍历，即可得到中序遍历的逆序，同时我们可以不使用List保存遍历结果，而是找到了直接返回，空间复杂度从O(N)降低到O(1)
二叉搜索树第 k 大的节点可转化为求此树的中序遍历的倒序的第 k 个节点

class Solution {
    int ans = 0, count = 0;
    public int kthLargest(TreeNode root, int k) {
        inorder(root, k);
        return ans;
    }
    void inorder(TreeNode root, int k){
        if (root == null)   return;
        // 右
        inorder(root.right, k);
        // 根
        if (++count == k){
            ans = root.val;
            // 提前结束
            return;
        }
        // 左
        inorder(root.left, k);
    }
}

迭代版本：

class Solution {
    public int kthLargest(TreeNode root, int k) {
        Deque<TreeNode> stack = new LinkedList<>();
        while (!stack.isEmpty() || root != null){
            while(root != null){
                stack.push(root);
                root = root.right;    
            }
            root = stack.peek();
            stack.pop();
            if (--k == 0)   return root.val;
            root = root.left;
        }
        return 0;
    }
}