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

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

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

示例 1:

示例 2:

限制：

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

解法一：

按右根左的顺序递归遍历二叉搜索树，第k个结点，即使第k大的结点

class Solution {
    public int kth = 0;
    public int k = 0;
    // 递归中序遍历,右根左，第k个结点，即使第k大的结点
    public int kthLargest(TreeNode root, int k) {
        this.k = k;
        midTravesal(root);
        return kth;
    }

    public void midTravesal(TreeNode root){

        if(root != null){
            midTravesal(root.right);
            k--;
            if(k == 0){
                kth = root.val;
                return;
            }   
            midTravesal(root.left);
        }
    }
}

leetcode运行时间为0ms, 空间为38.8mb

复杂度分析：

时间复杂度：遍历了树的k个结点，所以时间复杂度为O(k)

空间复杂度：递归栈的深度就是空间复杂度，二叉搜索树的最大深度为O(n), 最小深度为O(logn)

解法二：

利用栈的方式实现非递归访问栈，同样访问的第k个结点是我们的目标结点

class Solution {

    // 利用栈实现中序遍历,右根左，第k个结点，即使第k大的结点
    public int kthLargest(TreeNode root, int k) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode top = root;
        while(!stack.isEmpty() || top != null){
            while(top != null){
                stack.push(top);
                top = top.right;
            }
            top = stack.pop();
            k--;
            if(k == 0){
                return top.val;
            }
            top = top.left;
        }
        return 0;
    }
}

leetcode运行时间为1ms, 空间为38.5mb

复杂度分析：

时间复杂度：遍历了树的k个结点，所以时间复杂度为O(k)

空间复杂度：栈的大小就是空间复杂度，所以同样是O(k)