【一天一大 lee】 二叉搜索树中的众数 (难度:简单)-Day20200924
【一天一大 lee】 二叉搜索树中的众数 (难度:简单)-Day20200924
前端小书童
发布于 2020-09-29 20:21:47
发布于 2020-09-29 20:21:47
题目:[1]
给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）。
假定 BST 有如下定义：
- 结点左子树中所含结点的值小于等于当前结点的值
- 结点右子树中所含结点的值大于等于当前结点的值
- 左子树和右子树都是二叉搜索树
例如：给定 BST [1,null,2,2],
示例：
1
\
2
/
2返回[2].
提示: 如果众数超过 1 个，不需考虑输出顺序
进阶: 你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）
抛砖引玉
思路
先不考虑进阶中不使用额外空间的逻辑：
- 声明一个 map 记录每个节点出现的次数,并且记录最大出现次数
- 遍历二叉树(DFS 或者 BFS)
- 遍历 map 找出频率等于最大出现次数的数
/**
* 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 map = new Map(),
maxNum = 0,
_result = []
function dfs(node) {
if (node === null) return
if (map.has(node.val)) {
map.set(node.val, map.get(node.val) + 1)
} else {
map.set(node.val, 1)
}
maxNum = Math.max(maxNum, map.get(node.val))
dfs(node.left)
dfs(node.right)
}
// 遍历二叉树记录每个数字出现次数
dfs(root)
for (let [key, value] of map) {
if (maxNum === value) _result.push(key)
}
return _result
}加上进阶的限制的不使用额外的空间：
因为题目指定了二叉树的类型是二叉搜索树，二叉搜索树的特性就是中序遍历元素是递增的。
那么中序遍历传入的二叉树，重复出现的元素一定是相邻出现的，那么记录：
- 当前出现最高频率：maxCount
- 后面遇到的新节点：nextNode
- 后面遇到的新节点出现的频率：nextNum
当 nextNum 大于 maxCount 时则 nextNode 替换结果数组中的元素
当 nextNum 等于 maxCount 时则 nextNode 追加到结果数组中
中序遍历
var findMode = function(root) {
let maxNum = 0,
nextNode = null,
nextNum = 0,
_result = []
function dfs(node) {
if (node === null) return
dfs(node.left)
update(node.val)
dfs(node.right)
}
// 遍历二叉树记录每个数字出现次数
dfs(root)
function update(val) {
if (val === nextNode) {
++nextNum
} else {
nextNum = 1
nextNode = val
}
if (nextNum === maxNum) {
_result.push(nextNode)
}
if (nextNum > maxNum) {
maxNum = nextNum
_result = [nextNode]
}
}
return _result
}Morris 中序遍历
Morris 中序遍历：
- 如果当前节点的左子节点为空，处理当前节点，并遍历当前节点的右子节点；
- 如果当前节点的右子节点不为空，找到当前节点左子树的最右节点（该节点为当前节点中序遍历的前驱节点）；
- 如果最右节点的右指针为空，将最右节点的右指针指向当前节点，遍历当前节点的左子节点；
- 如果最右节点的右指针不为空，将最右节点的右指针重新置为空（恢复树的原状），处理当前节点，并将当前节点置为其右节点；重复步骤 1 和步骤 2，直到遍历结束。
var findMode = function(root) {
let maxNum = 0,
nextNode = null,
nextNum = 0,
_result = []
let node = root,
prevNode = null
while (node !== null) {
if (node.left === null) {
update(node.val)
node = node.right
continue
}
prevNode = node.left
while (prevNode.right !== null && prevNode.right !== node) {
prevNode = prevNode.right
}
if (prevNode.right === null) {
prevNode.right = node
node = node.left
} else {
prevNode.right = null
update(node.val)
node = node.right
}
}
function update(val) {
if (val === nextNode) {
++nextNum
} else {
nextNum = 1
nextNode = val
}
if (nextNum === maxNum) {
_result.push(nextNode)
}
if (nextNum > maxNum) {
maxNum = nextNum
_result = [nextNode]
}
}
return _result
}本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2020-09-24，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录