剑指 Offer 54. 二叉搜索树的第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 ≤ 二叉搜索树元素个数
遍历并排序
思路
要找到第k大的值,我们一定首先要遍历所有元素,得到所有结果然后进行比较
我们通过层序遍历二叉树
将没一个节点放入stack中,并将每一个节点的值放在一个数组res中
如果还有左子树或者右子树,则继续放入stack中继续遍历
最后得到res是所有节点值的集合
要找到第k大的元素,还需要对结果res进行排序
这里我们直接用sort来进行一个降序排序
最后直接读取数组的第(k-1)坐标的值就是第k大的值
var kthLargest = function (root, k) {
var stack = [root]
var res = []
while (stack.length) {
var item = stack.pop()
res.push(item.val)
if (item.left) {
stack.push(item.left)
}
if (item.right) {
stack.push(item.right)
}
}
res.sort((a, b) => b - a)
return res[k - 1]
};