代码随想录算法训练营第十九天（二叉树篇）| 501. 二叉搜索树中的众数

501. 二叉搜索树中的众数

题目链接：501. 二叉搜索树中的众数 - 力扣（LeetCode）

思路：用中序遍历二叉搜索树，用字典记录每个节点出现的频率，返回字典中频率（value)最大的节点值(key)。

class Solution(object):
    def __init__(self):
        self.freq = dict()
        self.pre = None

    def traversal(self, cur):
        if not cur:
            return None
        self.traversal(cur.left)
        if cur.val not in self.freq:
            self.freq[cur.val] = 1
        if self.pre:          
            if self.pre.val == cur.val:
                self.freq[self.pre.val] = self.freq.get(self.pre.val) + 1
        self.pre = cur  # Update self.pre for the next iteration
        self.traversal(cur.right)


    def findMode(self, root):
        self.freq = dict()
        self.pre = None
        self.traversal(root)
        result = []
        max_freq = max(self.freq.values())
        for key, freq in self.freq.items():
            if freq == max_freq:
                result.append(key)
        return result

这次递归的思路比较顺，反而在字典的应用中花了点时间，好久没用，有些生疏了。

table = dict()  # 字典形式{key1: value1, key2:value2, ...}
table[key] = value # 添加字典的key和对应的值。 注意是中括号，之前总会习惯打成大括号
table.get(key)  # 查询字典的kay所对应的值
max_freq = max(table.values())   # 找到字典中key对应的最大的value
# 返回所有zuida的value对应的key
for key, freq in self.freq.items():
    if freq == max_freq:
        result.append(key)