剑指 54. 二叉搜索树的第k大节点 - 难度简单

1. 题目描述

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

示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4

2
输出: 4

示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof

2. 题解

中序遍历倒序 - 右根左

看到题目想到了是中序遍历的倒序，但是对于变量 count 和 res 是作为参数还是返回值没弄明白，看到别人给出的答案是设置为类的成员变量。不过做题还是建议少用设置成员变量的方式。（这道题目前还没有想到怎么避免使用成员变量）。

class Solution {
	//声明两个成员变量
    private int count = 0, res = 0;
    public int kthLargest(TreeNode root, int k) {
        count = k;
        dfs(root);
        return res;
    }

    private void dfs(TreeNode root){
        if(root == null || count == 0){
            return;
        }
        //右
        dfs(root.right);
        //根
        if(--count == 0){
            res = root.val;
            return;
        }
        //左
        dfs(root.left);
    }
}