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