leetcode:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/submissions/
描述:
给定一棵二叉搜索树,请找出其中第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
思路:
二叉排序数的中序遍历是有序的。可以利用这一性质,题目中求的是第k 大,所以可以倒序输出,如果是第k小,就可以正序输出。
下面代码中的思路是参考大佬的。很巧妙的一点是他把中序遍历:左-根-右,变成了中序倒序:右-根-左。而且我一开始想的是需要一个list来盛树中的节点,结果大佬在中序遍历中直接提前返回。太妙了。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
// LinkedList<Integer> list = new LinkedList<>();
int res,k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
order(root);
return res;
}
public void order(TreeNode root){
if(root == null) return;
order(root.right);
if(--k == 0){ //判断
res = root.val;
return; //提前返回
}
order(root.left);
}
}