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

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

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

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

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

示例 :

输入：s = "dcab", pairs = [[,],[,]]

输出："bacd"

解释：

交换 s[] 和 s[], s = "bcad"

交换 s[] 和 s[], s = "bacd"

示例 ：

输入：s = "dcab", pairs = [[,],[,],[,]]

输出："abcd"

解释：

交换 s[] 和 s[], s = "bcad"

交换 s[] 和 s[], s = "acbd"

交换 s[] 和 s[], s = "abcd"

示例 ：

输入：s = "cba", pairs = [[,],[,]]

输出："abc"

解释：

交换 s[] 和 s[], s = "bca"

交换 s[] 和 s[], s = "bac"

交换 s[] 和 s[], s = "abc"

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

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

class Solution {

public:

    int p[] = {-}; //存储每个点的祖宗节点

    // 返回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 = ; i <= s.size(); i ++ ) p[i] = i;

        for(int i = ; i  < pairs.size();i++){

            int a = pairs[i][];

            int b = pairs[i][];

            p[find(a)] = find(b);

        }

        //================================

        int n = s.size();

        vector<vector<int>> idx(n);

        vector<string> ss(n);

        for(int i = ; i < n;i++){

            int id = find(i);

            idx[id].push_back(i);

            ss[id].push_back(s[i]);

        }

        for(int i =; i < n;i++){

            sort(ss[i].begin(),ss[i].end());

            for(int k = ;k < idx[i].size();k++){

                s[idx[i][k]] = ss[i][k];

            }

        }

        return s;

    }

};