652. 寻找重复的子树

652. 寻找重复的子树

题目描述

给定一棵二叉树，返回所有重复的子树。对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。
两棵树重复是指它们具有相同的结构以及相同的结点值。
示例 1：

下面是两个重复的子树：

和

思路

使用深度优先搜索，其中递归函数返回当前子树的序列化结果。把每个节点开始的子树序列化结果保存在 map 中，然后判断是否存在重复的子树。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
        vector<TreeNode*> res;
        if(root == NULL){
            return res;
        }
        unordered_map<string,int> m;
        dfs(root,res,m);
        return res;
    }
    string dfs(TreeNode* root, vector<TreeNode*>& res,unordered_map<string, int>& m){
        if(root == NULL){
            return "#";
        }
        string key = to_string(root->val) + "," + dfs(root->left,res,m) + "," + dfs(root->right,res,m);
        if(++m[key] == 2){
            res.push_back(root);
        }
        return key;
    }
};

复杂度分析

时间复杂度：O(N^2 )，其中 N 是二叉树上节点的数量。遍历所有节点，在每个节点处序列化需要时间 O(N)。
空间复杂度：O(N^2)。