[LeetCode] 1202. Smallest String With Swaps

You are given a string s, and an array of pairs of indices in the string pairs where pairs[i] = [a, b] indicates 2 indices(0-indexed) of the string.

You can swap the characters at any pair of indices in the given pairs any number of times.

Return the lexicographically smallest string that s can be changed to after using the swaps.

Example 1:

Input: s = "dcab", pairs = [[0,3],[1,2]]
Output: "bacd"
Explaination: 
Swap s[0] and s[3], s = "bcad"
Swap s[1] and s[2], s = "bacd"

Example 2:

Input: s = "dcab", pairs = [[0,3],[1,2],[0,2]]
Output: "abcd"
Explaination: 
Swap s[0] and s[3], s = "bcad"
Swap s[0] and s[2], s = "acbd"
Swap s[1] and s[2], s = "abcd"

Example 3:

Input: s = "cba", pairs = [[0,1],[1,2]]
Output: "abc"
Explaination: 
Swap s[0] and s[1], s = "bca"
Swap s[1] and s[2], s = "bac"
Swap s[0] and s[1], s = "abc"

Constraints:

• 1 <= s.length <= 10^5
• 0 <= pairs.length <= 10^5
• 0 <= pairs[i][0], pairs[i][1] < s.length
• s only contains lower case English letters.

交换字符串中的元素。

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-string-with-swaps
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题意不难理解，题目给出的 pairs[i] 总能使某两个索引对相连，我们可以这样想，如果索引对多了，是不是有可能形成好几组索引对，每一组里面是多个索引相互关联的？比如题目给的第三个例子，0 - 1，1 - 2，那么我们可以得出 0，1，2 这三个索引是一组的。按照这样的思路，我们可以把解法往并查集上靠拢。

对于 input 字符串里的每一个字母，首先我们需要找到其在并查集里面的祖先，记为 parent[i]，这样对于每一个 index，我们都记录了它的父节点的 index。对于 input 给定的每一个 pairs 而言，我们通过 union find 确定一下这一对 index 到底谁是谁的父节点。此时我们再次遍历字符串的长度 len，对于遍历到的每个 index，因为之前通过 parent[i] 确认了每个 index 的父级 index 是谁，所以此时我们可以把父节点相同的字母通过 priority queue 放在一起。最后一轮再次遍历长度 len，这样我们可以通过 hashmap 找到每一组索引里面字典序最小的那个，直到把整个字符串都拼接好。

时间O(V + E) - 其实是瞎写的~~

空间O(V + E) - 其实是瞎写的~~

Java实现

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

        for (List<Integer> pair : pairs) {
            int ancestry1 = find(pair.get(0), parent);
            int ancestry2 = find(pair.get(1), parent);
            parent[ancestry2] = ancestry1;
        }

        HashMap<Integer, PriorityQueue<Character>> map = new HashMap<>();
        for (int i = 0; i < len; i++) {
            int ancestry = find(i, parent);
            if (!map.containsKey(ancestry)) {
                map.put(ancestry, new PriorityQueue<>());
            }
            map.get(ancestry).offer(s.charAt(i));
        }

        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < len; i++) {
            PriorityQueue<Character> queue = map.get(find(i, parent));
            sb.append(queue.poll());
        }
        return sb.toString();
    }

    private int find(int i, int[] parent) {
        if (parent[i] != i) {
            parent[i] = find(parent[i], parent);
        }
        return parent[i];
    }
}

 

LeetCode 题目总结