LeetCode アルゴリズム_C++ - 重複するサブツリーの検索

バイナリ ツリーのルート ノード ルートを指定すると、重複するサブツリーをすべて返します。

同じタイプの重複サブツリーの場合は、それらのいずれかのルート ノードを返すだけで済みます。

2 つのツリーが同じ構造と同じノード値を持つ場合、それらのツリーは重複しているとみなされます。

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