1. 程式人生 > 實用技巧 >20210111 交換字串中的元素(並查集)

20210111 交換字串中的元素(並查集)

給你一個字串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" 來源:力扣(LeetCode) 連結:https://leetcode-cn.com/problems/smallest-string-with-swaps 著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
public String smallestStringWithSwaps(String s, List<List<Integer>> pairs) { }

思路:依次按pairs下標對比字串大小 字典序小的排前面 [[0,3],[1,2],[0,2]]

(按前面做題的思路仔細想這個思路不行 因為按下標pairs[0]排好序後在排pairs[1], pairs[1]若包含pairs[0]就可能把pairs[0]排序打亂)

有點難

思路:並查集

就是把下標pairs中有交集的下標集合都整合成一個集合如:[[0,3],[1,2][0,4]] 可以當成(0,3,4)(1,2)的2個集合

find(int x)返回彙總的集合的下標

下面大佬解答

class Solution {

    int[] p;

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

    public String smallestStringWithSwaps(String s, List<List<Integer>> pairs) {
        p = new int[s.length()];
        for (int i = 0; i < s.length(); i++) {
            p[i] = i;
        }

        for (List<Integer> pair : pairs) {
            int a = pair.get(0), b = pair.get(1);
            p[find(a)] = find(b);
        }

        StringBuilder sb = new StringBuilder();

        HashMap<Integer, PriorityQueue<Character>> map = new HashMap<>();

        for (int i = 0; i < s.length(); i++) {
            int r = find(i);
            if (map.get(r) == null){
                int finalI = i;
                map.put(r,new PriorityQueue<Character>(){{offer(s.charAt(finalI));}});
            }else {
                map.get(r).offer(s.charAt(i));
            }
        }

        for (int i = 0; i < s.length(); i++) {
            Character poll = map.get(find(i)).poll();
            if (poll!=null) sb.append(poll);
        }

        return sb.toString();
    }

}