力扣（LeetCode）算法_C++——寻找重复的子树

吴俊荣2023-09-07 8:55

给你一棵二叉树的根节点 root ，返回所有重复的子树。

对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。

如果两棵树具有相同的结构和相同的结点值，则认为二者是重复的。

示例 1：

输入：root = [1,2,3,4,null,2,4,null,null,4]

输出：[[2,4],[4]]

示例 2：

输入：root = [2,1,1]

输出：[[1]]

示例 3：

输入：root = [2,2,2,3,null,3,null]

输出：[[2,3],[3]]

cpp 复制代码

class Solution {
public:
    vector<TreeNode*> findDuplicateSubtrees(TreeNode* root) {
        dfs(root);
        return {repeat.begin(), repeat.end()};
    }

    string dfs(TreeNode* node) {
        if (!node) {
            return "";
        }
        string serial = to_string(node->val) + "(" + dfs(node->left) + ")(" + dfs(node->right) + ")";
        if (auto it = seen.find(serial); it != seen.end()) {
            repeat.insert(it->second);
        }
        else {
            seen[serial] = node;
        }
        return serial;
    }

private:
    unordered_map<string, TreeNode*> seen;
    unordered_set<TreeNode*> repeat;
};