面试题54. 二叉搜索树的第k大节点(简单，常考)

题目

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

思路

中序遍历二叉搜索树，是会按照从大到小(右中左)或从小到大(左中右)的顺序打印出二叉搜索树中的值。

那我们就在第k次将结果保留起来就好了，当过了第k次以后，后面的遍历就不需要了。

易错点：k在递归的过程中是全局变量，不能写在递归函数里，因为k是针对整个函数，只进行k次

图片
作者：jyd
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/solution/mian-shi-ti-54-er-cha-sou-suo-shu-de-di-k-da-jie-d/
来源：力扣（LeetCode）

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);
    }
}

遍历

class Solution {

    int res, k;

    public int kthLargest(TreeNode root, int k) {
        if(root == null || k < 0) return -1;

        this.k = k;
        dfs(root);
        return res;
    }

    private void dfs(TreeNode root) {
        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode cur = root;
        while(!stack.isEmpty() || cur != null) {
            while(cur != null) {
                stack.push(cur);
                cur = cur.right;
            }
            if(k == 0) return;
            TreeNode node = stack.pop();
            k--;
            if(k == 0) res = node.val;
            if(node.left != null) {
                cur = node.left;
            }
        }
    }


    
}