leetcode-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”

// 并查集板子题
class Solution {
     
    public static String smallestStringWithSwaps(String s, List<List<Integer>> pairs) {
     
        int[] arr = new int[s.length()];
        for (int i = 0; i < arr.length; i++) {
     
            arr[i] = i;
        }
        for (List<Integer> pair : pairs) {
     
            int a = find(arr, pair.get(0));
            int b = find(arr, pair.get(1));
            if (a < b) {
     
                arr[b] = a;
            } else {
     
                arr[a] = b;
            }
        }
        // 把一组分一起
        Map<Integer, Queue<Character>> map = new HashMap<>();
        for (int i = 0; i < s.length(); i++) {
     
            map.computeIfAbsent(find(arr,arr[i]), k -> new PriorityQueue<>()).offer(s.charAt(i));
        }
        // 从每个组内获取最小值
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < s.length(); i++) {
     
            sb.append(map.get(find(arr,arr[i])).poll());
        }
        return sb.toString();
    }

    // 合并
    private static void join(int[] arr, int a, int b) {
     
        int x=find(arr,a);
        int y=find(arr,b);
        if(x<y) {
     
            arr[y]=x;
        }else {
     
            arr[x]=y;
        }
    }
    // 查找
    private static int find(int[] arr, int a) {
     
        while(a!=arr[a]) {
     
            a=arr[a];
        }
        return a;
    }
}