给你一个字符串 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”
并查集的题目,还是有些难度的。
用并查集保存下标,可以交换的下标是在同一个联通分量中,排序同一个联通分量中的字符的操作比较有趣:
mmap中保存:<根,可以交换的下标们>,这个匿名函数运用的很好:
sort(v.begin(), v.end(), [&](int i, int j)->bool {
return s[i] < s[j];
class UnionFind {
unordered_map<int, int> mmap;
public:
void merge(int s1, int s2) {
int r1 = find(s1);
int r2 = find(s2);
if (r1 != r2) {
mmap[r1] = r2;
}
}
int find(int s) {
int root = s;
while (mmap[root] != root) {
root = mmap[root];
}
while (mmap[s] != s) {
int tmp = mmap[s];
mmap[s] = root;
s = tmp;
}
return root;
}
void add (int s) {
if (mmap.count(s) == 0) {
mmap[s] = s;
}
}
bool isConnected(int s1, int s2) {
int r1 = find(s1);
int r2 = find(s2);
if (r1 != r2) {
return false;
}
return true;
}
};
class Solution {
public:
string smallestStringWithSwaps(string s, vector<vector<int>>& pairs) {
UnionFind uf;
for (int i = 0; i < s.size(); i++) {
uf.add(i);
}
// 先建立起来
for (const auto& pair : pairs) {
uf.merge(pair[0], pair[1]);
}
unordered_map<int, vector<int>> mmap;
for (int i = 0; i < s.size(); i++) {
mmap[uf.find(i)].push_back(i);
}
string res(s.size(),'.');
for (auto pair :mmap) {
vector<int> v = pair.second;
// v中保存的是排序好的字符
sort(v.begin(), v.end(), [&](int i, int j)->bool {
return s[i] < s[j];
});
// pair.second保存的是字符的位置,将对应的位置上放上排序好的字符即可
for (int i = 0; i < v.size(); i++) {
res[pair.second[i]] = s[v[i]];
}
}
return res;
}-*
/.
};