剑指offer54 二叉搜索树的第k大节点

题目

给定一棵二叉搜索树，请找出其中第 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 ≤ 二叉搜索树元素个数

方法

回溯法

• 时间复杂度：O（n）当二叉搜索树退化成链表时，最坏情况是n
• 空间复杂度：O（n）

class Solution {
    private int ans = -1;
    private int count = 0;
    public int kthLargest(TreeNode root, int k) {
        dfs(root,k);
        return ans;
    }
    private void dfs(TreeNode root,int k){
        if(root==null) return;
        dfs(root.right,k);
        count++;
        if(count>k) return;
        if(count==k) {
            ans = root.val;
            return;
        }
        dfs(root.left,k);
    }
}