【一天一大 lee】二叉搜索树中的众数 (难度:简单)-Day20200924

【公众号：前端小书童】

题目:

给定一个有相同值的二叉搜索树（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. 如果当前节点的右子节点不为空，找到当前节点左子树的最右节点（该节点为当前节点中序遍历的前驱节点）；
   • 如果最右节点的右指针为空，将最右节点的右指针指向当前节点，遍历当前节点的左子节点；
   • 如果最右节点的右指针不为空，将最右节点的右指针重新置为空（恢复树的原状），处理当前节点，并将当前节点置为其右节点；
3. 重复步骤 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
}