力扣(LeetCode)算法_C++——寻找重复的子树

给你一棵二叉树的根节点 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]]

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;
};

猜你喜欢

转载自blog.csdn.net/weixin_43945471/article/details/132724465