方法一:字符串保存
对每个节点为根的子树进行字符串序列化操作
def collect(node):
serial = "{},{},{}".format(node.val, collect(node.left), collect(node.right))
return serial
再使用dict进行存储
m = defaultdict(int)
m[serial]+=1
方法二:为每个根生成标识符
使用defaultdict的default_factory参数生成
m = defaultdict()
m.default_factory = m.__len__
再使用counter记录每个子树的次数
c = counter()
tree_id = m[node.val,lookup(node.left),lookup(node.right)]
c[tree_id]+=1
思路来源:
https://leetcode-cn.com/problems/find-duplicate-subtrees/