给定一棵二叉树,返回所有重复的子树。对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。

两棵树重复是指它们具有相同的结构以及相同的结点值。

思路

首先是进行序列化,然后序列化的过程中,进行树的深度优先搜索,在树的搜索过程中,进行一个结果的统计。
毕竟树的深度优先访问的时间负杂度是O(n)的。

代码

class Solution {
   
public:
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
   
        if (root == nullptr) return {
   };
        vector<TreeNode*> res;
        map<string, int> m;
        dfs(root, res, m);
        return res;
    }
    string dfs(TreeNode* root, vector<TreeNode*>&res, map<string,int>&m) {
   
        if (root == nullptr) return "";
        string ks = to_string(root->val) + "," + dfs(root->left,res,m) + "," + dfs(root->right, res, m); 
        if (++m[ks] == 2) {
   
            res.push_back(root);
        }
        return ks;
    }
};

总结

树的遍历过程+操作就可以完成很多任务了。