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

作者:Choi Yang

更新于:3 个月前

字数统计:529 字

阅读时长:1 分钟

阅读量:

题目描述

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

假定 BST 有如下定义：

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

例如：

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

   1
    \
     2
    /
   2
返回[2].

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

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

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

解题思路

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

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

/**
 * 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;
};

学如逆水行舟，不进则退

Contributors

Choi Yang