[路飞]_leetcode-剑指 Offer 54-二叉搜索树的第k大节点

[题目地址] [B站地址]

给定一棵二叉搜索树,请找出其中第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 个节点,返回节点值即可

动画演示如下:

leetcode-剑指 Offer 54-二叉搜索树的第k大节点.gif

代码如下:

var kthLargest = function(root, k) {
  // 初始化 num
  let num = 0;
  // 右根左顺序遍历二叉搜索树
  function getNodeVal(root){
    if(root === null) return;
    let right = getNodeVal(root.right);
    if(right!==undefined) return right;
    num++;
    // 如果 num = k,说明当前节点为第k个节点
    if(num===k) return root.val;
    return getNodeVal(root.left);
  }
  // 返回结果
  return getNodeVal(root);
};
复制代码

至此我们就完成了 leetcode-剑指 Offer 54-二叉搜索树的第k大节点

如有任何问题或建议,欢迎留言讨论!

猜你喜欢

转载自juejin.im/post/7040444357278711838