LeetCode 1202.交换字符串中的元素 C++

LeetCode 1202.交换字符串中的元素 C++

给你一个字符串 s，以及该字符串中的一些「索引对」数组 pairs，其中 pairs[i] = [a, b] 表示字符串中的两个索引（编号从 0 开始）。

你可以 任意多次交换 在 pairs 中任意一对索引处的字符。

返回在经过若干次交换后，s 可以变成的按字典序最小的字符串。

示例 1:

输入：s = “dcab”, pairs = [[0,3],[1,2]]
输出：“bacd”
解释：
交换 s[0] 和 s[3], s = “bcad”
交换 s[1] 和 s[2], s = “bacd”

示例 2：

输入：s = “dcab”, pairs = [[0,3],[1,2],[0,2]]
输出：“abcd”
解释：
交换 s[0] 和 s[3], s = “bcad”
交换 s[0] 和 s[2], s = “acbd”
交换 s[1] 和 s[2], s = “abcd”

示例 3：

输入：s = “cba”, pairs = [[0,1],[1,2]]
输出：“abc”
解释：
交换 s[0] 和 s[1], s = “bca”
交换 s[1] 和 s[2], s = “bac”
交换 s[0] 和 s[1], s = “abc”

提示：

• 1 <= s.length <= 10^5
• 0 <= pairs.length <= 10^5
• 0 <= pairs[i][0], pairs[i][1] < s.length
• s 中只含有小写英文字母

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-string-with-swaps

问题分析： 很明显是图问题，pairs形成的环里的元素是可以任意交换的，所以我们只需找到环里的元素对应的下标然后进行排序就行了，典型的并查集。

代码如下：

//并查集模板
class UnionFind{
private:
    vector<int> parent;
    vector<int> rank;
public:
    UnionFind(int n){
        parent.resize(n);
        rank.resize(n);
        for(int i=0;i<n;i++){
            parent[i]=i;
            rank[i]=1;
        }
    }
    void Union(int x,int y){
        int x_root=Find(x);
        int y_root=Find(y);
        if(rank[x_root]==rank[y_root]){
            parent[x_root]=y_root;
            rank[y_root]++;
        }
        else if(rank[x_root]<rank[y_root]){
            parent[x_root]=y_root;
        }
        else{
            parent[y_root]=x_root;
        }
    }
    int Find(int x){
        if(x!=parent[x]){
            x=Find(parent[x]);
        }
        return x;
    }
};
class Solution {
public:
    string smallestStringWithSwaps(string s, vector<vector<int>>& pairs) {
        if(s.length()==0){
            return 0;
        }
        int len=s.length();
        UnionFind unionFind(len);
        for(vector<int>& pair:pairs){
            unionFind.Union(pair[0],pair[1]);
        }
        //这里采用优先队列进行储存环里的元素，自动排序
        unordered_map<int,priority_queue<char,vector<char>,greater<char>>> hashtable;
        for(int i=0;i<len;i++){
            int root=unionFind.Find(i); //找到该元素对应的根
            hashtable[root].push(s[i]); //将该元素放入对应的根里
        }
        string res;
        for(int i=0;i<len;i++){
            int root=unionFind.Find(i);
            res+=hashtable[root].top();
            hashtable[root].pop();
        }
        return res;
    }
};