[路飞]_程序员必刷力扣题: 剑指 Offer 54. 二叉搜索树的第k大节点

249

剑指 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]
};