[算法练习及思路-leetcode剑指offer(Java解法)]No55.二叉搜索树的第k大节点

题号:剑指 Offer 54

题目名:二叉搜索树的第k大节点

原题URL:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/

题目描述

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

思路

1.二叉搜索树的中序遍历就是从小到大的顺序

2.先按顺序放到list,然后逆序取出来

解题代码

public class Solution {
    List<Integer> list = new ArrayList<>();
    public int kthLargest(TreeNode root, int k) {
        //中序遍历是从到大的顺序
        dfs(root);
        for (int i = list.size() - 1; i >= 0; i--) {
            k--;
            if(k==0) return list.get(i);
        }
        return 0;
    }

    private void dfs(TreeNode root) {
        if(root.left!=null) dfs(root.left);
        list.add(root.val);
        if(root.right!=null) dfs(root.right);
    }
}