LeeCode 1984 Trie + 序列化

题意

传送门 LeeCode 1948 删除系统中的重复文件夹

题解

将文件系统建为 $Trie$，设根节点深度为 $0$，此时有 ${\sum }_{u\in Trie}u(depth[u]+1)\le 2\times 10^5$。

考虑将子文件夹，即某棵子树序列化。具体而言，后序遍历 $Trie$，设以 $u$ 为根节点的子树，按照与孩子节点 $ch$ 连边所代表的字符串 $s_{ch}$ 使孩子节点有序，那么可以自定义序列化为
$$serial(u)=(s_{c{h}_0},serial(c{h}_0))(s_{c{h}_1},serial(c{h}_1))\cdots )$$ 序列化的子树中包含了子树的信息，可以通过反序列化得到原来的子树。序列化类似于哈希的思想，且保证哈希值对于不同结构子树不会产生冲突。

任一节点仅会出现在其本身与其祖先节点的序列化表示中，那么所有节点的序列化表示规模为 ${\sum }_{u\in Trie}u(depth[u]+1)\le 2\times 10^5$。统计序列化表示出现的次数，通过遍历 $Trie$ 统计答案即可。

class Solution
{
    static const int maxn = 200005;

public:
    int rt, tot;
    map<string, int> trie[maxn];
    string rec[maxn];
    unordered_map<string, int> cnt;
    void insert(vector<string> &vec)
    {
        int p = rt;
        for (auto &u : vec)
        {
            if (trie[p].count(u) == 0)
                trie[p][u] = tot++;
            p = trie[p][u];
        }
    }
    string get_serial(int p)
    {
        string res = "";
        for (auto &pr : trie[p])
            res += '(' + pr.first + ',' + get_serial(pr.second) + ')';
        ++cnt[res];
        return rec[p] = res;
    }
    void dfs(int p, vector<string> &tmp, vector<vector<string>> &res)
    {
        if (cnt[rec[p]] > 1 && rec[p].size() > 0)
            return;
        if (p != rt)
            res.push_back(tmp);
        for (auto &pr : trie[p])
        {
            tmp.push_back(pr.first);
            dfs(pr.second, tmp, res);
            tmp.pop_back();
        }
    }
    vector<vector<string>> deleteDuplicateFolder(vector<vector<string>> &paths)
    {
        tot = 0, rt = tot++;
        for (auto &v : paths)
            insert(v);
        get_serial(rt);
        vector<vector<string>> res;
        vector<string> tmp;
        dfs(rt, tmp, res);
        return res;
    }
};