leetcode 5199. 交换字符串中的元素

地址  https://leetcode-cn.com/contest/weekly-contest-155/problems/smallest-string-with-swaps/

给你一个字符串 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"

 

这个使用的并查集用来解决 在一类的索引查找

后面字符串的排序和索引排序 需要注意下

class Solution {
public:
    int p[100010] = {-1}; //存储每个点的祖宗节点

    // 返回x的祖宗节点
    int find(int x)
    {
        if (p[x] != x) p[x] = find(p[x]);
        return p[x];
    }
    
    string smallestStringWithSwaps(string s, vector<vector<int>>& pairs) {
        for (int i = 0; i <= s.size(); i ++ ) p[i] = i;
        for(int i = 0; i  < pairs.size();i++){
            int a = pairs[i][0];
            int b = pairs[i][1];
            p[find(a)] = find(b);
        }
        
        //================================
        int n = s.size();
        vector<vector<int>> idx(n);
        vector<string> ss(n);
        
        for(int i = 0; i < n;i++){
            int id = find(i);
            idx[id].push_back(i);
            ss[id].push_back(s[i]);
        }
        
        for(int i =0; i < n;i++){
            sort(ss[i].begin(),ss[i].end());
            for(int k = 0;k < idx[i].size();k++){
                s[idx[i][k]] = ss[i][k];
            }
        }
        
        
        
        return s;
    }
};