Leetcode 1202. 交换字符串中的元素 (转化为联通分量,独立排序,思维比较独特)

假设区间都是不相交的区间,此时区间彼此是独立的,因此只需要两两交换看交换后字典序是否会变大,此时是典序的贪心思路。如果出现区间相交的情况,由于可以进行任意多次交换,那么这组区域的字典序最小就是将他们排序。

因此思路是只需要找到所有联通区间,然后将其中对应字母排序,然后将排序后的结果插入到原来的字符串中,既可以得到重新组合后新生成的字符串。

 

统计区间的联通性,使用到的数据结构是并查集。并查集处理完后,用一个HashMap数据结构来保存,每个根节点对应的字符串映射,我们遍历一遍字符串,将这个字符映射建立起来。然后我们遍历整个hashmap,对字符串排序,我们按照从大到小排序。这样在遍历一次字符串索引,找到其对应的根,在通过HashMap找到对应的字符串数组,然后取出最后一个,即为这个位置要放的字符,最后就可以完成。

class DisjointSetUnion {
private:
    vector<int> p;
    int n;

public:
    DisjointSetUnion(int _n) {
        n = _n;
        p.resize(n);
        for (int i = 0; i < n; i++) {
            p[i] = i;
        }
    }

    int find(int x) {
        if(p[x]!=x) p[x] = find(p[x]);
        return p[x];
    }

    void unionSet(int x, int y) {
        p[find(x)] = find(y);
    }
};

class Solution {
public:
    string smallestStringWithSwaps(string s, vector<vector<int>>& pairs) {
        DisjointSetUnion dsu(s.length());
        for (auto& it : pairs) {
            dsu.unionSet(it[0], it[1]);
        }
        unordered_map<int, vector<int>> mp;
        for (int i = 0; i < s.length(); i++) {
            mp[dsu.find(i)].emplace_back(s[i]);
        }
        for (auto& [x, vec] : mp) {
            sort(vec.begin(), vec.end(), greater<int>());    
        }
        for (int i = 0; i < s.length(); i++) {
            int x = dsu.find(i);
            s[i] = mp[x].back();                             
            mp[x].pop_back();
        }
        return s;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章