Skip to content
本页目录

LeetCode 501. 二叉搜索树中的众数

题目描述

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

  • 结点左子树中所含结点的值小于等于当前结点的值
  • 结点右子树中所含结点的值大于等于当前结点的值
  • 左子树和右子树都是二叉搜索树

例如:

javascript
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

提示:如果众数超过 1 个,不需考虑输出顺序

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/find-mode-in-binary-search-tree 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

由于 BST(二叉搜索树)的特殊性,我们采用递归来中序遍历,访问的节点值是有序的。然后重复节点,用计数器进行累加即可,如果有新值出现,则更新新值,然后计数器重置为 1。然后对于当前次数超过了最大值,则更新当前最大值,如果等于最大值,则代表出现了相同频率的数字,加入即可。

如果次数小于最大值,不需要什么操作。

javascript
/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var findMode = function (root) {
  let cnt = 0;
  let pre = 0;
  let res = [];
  let maxCnt = 0;
  let handle = (cur) => {
    // 相同的数,累加
    if (cur === pre) {
      cnt++;
    } else {
      // 有新数出现,重新置计数器为1,更新新数
      pre = cur;
      cnt = 1;
    }
    // 如果次数超过了最大值,更新当前最大值
    if (cnt > maxCnt) {
      maxCnt = cnt;
      res = [cur];
      // 如果有相同频率的数字出现,直接加入
    } else if (cnt === maxCnt) {
      res.push(cur);
    }
  };
  // 二叉搜索树,递归中序遍历方式
  let inOrder = (root) => {
    if (!root) return null;
    inOrder(root.left);
    handle(root.val);
    inOrder(root.right);
  };
  inOrder(root);
  return res;
};
javascript
学如逆水行舟,不进则退