【算法训练】二叉树系列（四）序列化及后序遍历的特殊位置

一、寻找重复的子树 652

1、分析

首先要比较结点是否相同，我们可以将该结点及其子树序列化为字符串保存并用于比较（采用前序或者后序结果都可以）。其次为了保证结果集中的相同结点只被加入一次，我们可以记录相同序列化字符串出现的次数，仅当第一次出现重复的时候加入结果集，其他情况不加。

注意到我们的关键代码都写在“后序位置”这个关键位置上，因为我们不仅需要传进来的参数信息，还需要知道子树通过函数返回值传递回来的数据。

具体见代码

2、代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:

    def findDuplicateSubtrees(self, root: Optional[TreeNode]) -> List[Optional[TreeNode]]:
        res = []
        dic = collections.Counter() #用于记录序列化字符串出现的次数
        def helper(node):
            if not node:
                return '#'
            temp = helper(node.left)+','+helper(node.right)+','+str(node.val) #后序序列化，注意到序列化的时候值之间要记得加分隔符，开始忘记了加逗号，导致两个节点的数字拼在一起，原本不等的序列化结果也错误相等了
            dic[temp] += 1
            if dic[temp]==2: #保证只往结果集里添加一次
                res.append(node)  
            return temp
        helper(root)
        return res

JS

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {TreeNode[]}
 */
var findDuplicateSubtrees = function(root) {
    let res = [],dic = {};
    var helper = function(node){
        if(node===null){
            return '#';
        }
        let temp = helper(node.left)+','+helper(node.right)+','+node.val;
        if(temp in dic){
            if(dic[temp]==1){
                res.push(node);
            }
            dic[temp] += 1;
        }else{
            dic[temp] = 1;
        }
        return temp;
    }
    helper(root);
    return res;

};