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
学如逆水行舟,不进则退