LeetCode 1202 Smallest String With Swaps

思路

首先要能想到并且发现，对于任何相连的，或者说在一棵树上pairs，比如[[0, 1], [0, 2], [1, 3], [1, 5], [2, 4], [2, 6]]，他们可以通过交换产生任意顺序的排列。证明的方法可以通过表示成树的形式然后分析。简单来说，对于任何两个一棵树上的pair[u, v]，都可以通过先将u交换到v，再反过来将v换到原来u的位置实现互换。例如[[0, 1], [0, 3]]，1和3互换可以通过swap(0, 1), swap(1, 3), swap(3, 0)来实现。

有了上述条件，需要做的就是找出所有在一棵树上的位置，然后把他们按字典序排列就可以了。dfs和并查集都可以实现找一棵树/一个集合的问题，按理说两个复杂度相差不是特别大，但是我写的DFS超时了emmm，所以就改成并查集了。

查找完所有一个集合的下标之后，将他们的字符拿出来，按字典序排序然后重新填入原来的下标即可，注意下标也要从小到大开始。

代码

class Solution {
public:
    int findFa(int x, vector<int> &fa) {
        return fa[x] == x ? x : fa[x] = findFa(fa[x], fa);
    }

    void unite(int x, int y, vector<int> &fa) {
        int fax = findFa(x, fa), fay = findFa(y, fa);
        fa[fay] = fax;
    }

    string smallestStringWithSwaps(string s, vector<vector<int>> &pairs) {
        const int sz = (int)s.size();
        vector<int> fa(sz, -1);
        for (int i = 0; i < sz; i++)
            fa[i] = i;
        for (auto &&p : pairs)
            unite(p[0], p[1], fa);
        unordered_map<int, vector<int>> fa2index;
        for (int i = 0; i < sz; i++)
            fa2index[findFa(i, fa)].push_back(i);
        for (auto &&p : fa2index) {
            vector<char> indexcp(p.second.size());
            for (size_t i = 0; i < p.second.size(); i++)
                indexcp[i] = s[p.second[i]];
            sort(indexcp.begin(), indexcp.end());
            for (size_t i = 0; i < indexcp.size(); i++)
                s[p.second[i]] = indexcp[i];
        }
        return s;
    }
};