【leetcode】【medium】【每日一题】1202. 交换字符串中的元素

1202. 交换字符串中的元素

给你一个字符串 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 中只含有小写英文字母

题目链接：https://leetcode-cn.com/problems/smallest-string-with-swaps/

思路

题目是关于联通的问题，且不要求回答路径，则可以判断是个并查集的题目。

大致思路是：先对所有pair对的联通关系用并查集进行记录，然后将每个分组的存储到各自的优先队列中，最后所求即为，依次输出s串中字符所属分组的队头。

代码写的有点乱，逻辑有可优化的地方。

class Solution {
public:
    string smallestStringWithSwaps(string s, vector<vector<int>>& pairs) {
        int num = s.size();
        if(num<=1) return s;
        int p_num = pairs.size();
        vector<priority_queue< char, vector<char>, greater<char>>> rec(num);
        vector<int> parent(num);
        for(int i=0; i<num; ++i){
            parent[i] = i;
        }
        for(int i=0; i<p_num; ++i){
            unionEle(pairs[i][0], pairs[i][1], parent);
        }
        for(int i=0; i<num; ++i){
            parent[i] = find(parent[i], parent);
            rec[parent[i]].push(s[i]);
        }
        string ans = "";
        for(int i=0; i<num; ++i){
            ans += (rec[parent[i]]).top();
            rec[parent[i]].pop();
        }
        return ans;
    }
    void unionEle(int p, int q, vector<int> &parent){
        int pp = find(p, parent);
        int pq = find(q, parent);
        parent[pq] = pp;
    }
    int find(int ele, vector<int> &parent){
        if(parent[ele]!=ele){
            parent[ele] = find(parent[ele], parent);
        }
        return parent[ele];
    }
};