题解 | #二叉树之寻找第k大#
二叉树之寻找第k大
https://www.nowcoder.com/practice/8e5f73fa3f1a407eb7d0b0d7a105805e
-
考察知识点: 中序遍历、递归、二叉搜索树
-
题目分析:
题目中要求出第k大的牛牛。实际上,对于一棵二叉搜索树,它的中序遍历是递增的。
因此,可以通过中序遍历得到二叉搜索树中的第k小的数。
此时反过来遍历,以右节点为前遍历即可得到递减的序列
-
所用编程语言:java
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @param k int整型
* @return int整型
*/
public int kthLargest (TreeNode root, int k) {
// write code here
Deque<TreeNode> deque = new ArrayDeque<>();
int res = root.val;
TreeNode temp = root;
while ((temp != null || !deque.isEmpty()) && k-- > 0) {
while (temp != null) {
deque.add(temp);
temp = temp.right;
}
temp = deque.pollLast();
res = temp.val;
temp = temp.left;
}
return res;
}
}