【二叉树】二叉搜索树中的众数

0x00 题目

给定一个有相同值的二叉搜索树（BST），找出 BST 中的所有众数（出现频率最高的元素）
如果众数超过 1 个，不需考虑输出顺序

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

---

0x01 思路

二叉搜索树的 中序遍历 是一个升序数组

可以把遍历结果放在一个 字典 里面
再遍历，根据 数量 来确定

时间复杂度: O(N)
空间复杂度: O(N)

---

此外
还可以顺序扫描【中序遍历】序列
用 base 记录当前的数字
用 count 记录当前数字重复的次数
用 maxCount 来维护已经扫描过的数当中，出现最多的那个数字的出现次数
用 res 数组记录出现的众数

时间复杂度: O(N)
空间复杂度: O(1)

---

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

中序遍历 解法：

func findMode(_ root: TreeNode?) -> [Int] {
    var res: [Int] = []
    var base = 0
    var count = 0
    var maxCount = 0
    
    func find(_ root: TreeNode?) {
        guard let root = root else { return }
        
        // 前序遍历位置
        find(root.left)
        
        // 中序遍历位置
        
        let val = root.val
        // 值一样，数量加1
        if val == base { count += 1}
        // 值不一样，重置
        else { count = 1; base = val }

        // 数量一样，添加
        if count == maxCount {
            res.append(val)
        }
        // 数量不一样，重置
        else if count > maxCount {
            maxCount = count
            res.removeAll()
            res.append(val)
        }
        
        find(root.right)
        // 后序遍历位置
    }
    
    find(root)
    return res
}

---

五笔学习

小程序：XWubi
配合公众号文章一起食用，效果更佳~
五笔教程之字根解析(四)

---