题目
给你两个整数数组 source 和 target ,长度都是 n 。还有一个数组 allowedSwaps ,其中每个 allowedSwaps[i] = [ai, bi] 表示你可以交换数组 source 中下标为 ai 和 bi(下标从 0 开始)的两个元素。注意,你可以按 任意 顺序 多次 交换一对特定下标指向的元素。
相同长度的两个数组 source 和 target 间的 汉明距离 是元素不同的下标数量。形式上,其值等于满足 source[i] != target[i] (下标从 0 开始)的下标 i(0 <= i <= n-1)的数量。
在对数组 source 执行 任意 数量的交换操作后,返回 source 和 target 间的 最小汉明距离 。
示例 1:
输入:source = [1,2,3,4], target = [2,1,4,5], allowedSwaps = [[0,1],[2,3]]
输出:1
解释:source 可以按下述方式转换:
- 交换下标 0 和 1 指向的元素:source = [2,1,3,4]
- 交换下标 2 和 3 指向的元素:source = [2,1,4,3]
source 和 target 间的汉明距离是 1 ,二者有 1 处元素不同,在下标 3 。示例 2:
输入:source = [1,2,3,4], target = [1,3,2,4], allowedSwaps = []
输出:2
解释:不能对 source 执行交换操作。
source 和 target 间的汉明距离是 2 ,二者有 2 处元素不同,在下标 1 和下标 2 。示例 3:
输入:source = [5,1,2,4,3], target = [1,5,4,2,3], allowedSwaps = [[0,4],[4,2],[1,3],[1,4]]
输出:0提示:
n == source.length == target.length
1 <= n <= 10^5
1 <= source[i], target[i] <= 10^5
0 <= allowedSwaps.length <= 10^5
allowedSwaps[i].length == 2
0 <= ai, bi <= n - 1
ai != bi
思路
做时看到有说是这个题的简单版本,就顺便来做一下这个题。
是需要将同一连通分量按字母顺序排列,这个题是需要对比
和
数字的次数,如果完全一致,就可以在交换顺序后达到最小的汉明距离。如果有不同的数字,那么就会增加汉明距离。因此,和1202不同的是,对于每一个连通分量代表元,我们统计
中各个数字的次数,然后遍历
去和
进行比对,具体的实现细节见代码注释。
代码
class Solution {
public int minimumHammingDistance(int[] source, int[] target, int[][] allowedSwaps) {
int n = source.length;
UnionFind uf = new UnionFind(n);
for (int[] swap : allowedSwaps) {
uf.union(swap[0], swap[1]);
}
// 统计target中,每个连通分量中各个数字出现的次数
// key为连通分量代表元,value为一个哈希表,存放连通分量各个数字出现的次数
Map<Integer, Map<Integer, Integer>> t = new HashMap<>();
for (int i = 0; i < n; i++) {
Map<Integer, Integer> map = t.computeIfAbsent(uf.find(i), k -> new HashMap<>());
map.put(target[i], map.getOrDefault(target[i], 0) + 1);
}
int diff = 0;
// 可以将target看做"标答",现在检验source有多符合标答
// 当然也可以像上面统计target一样,统计好source每个连通分量中各个数字出现的次数,这里省略了一个map
for (int i = 0; i < n; i++) {
Map<Integer, Integer> map = t.get(uf.find(i));
// source[i]在target中不一定存在,所以用了getOrDefault
int cnt = map.getOrDefault(source[i], 0);
// source[i]在target中找到了,将target中的次数减一
if (cnt > 0) {
map.put(source[i], cnt - 1);
} else {
// 没找到,就产生了一个汉明距离
diff++;
}
}
return diff;
}
}
class UnionFind {
int[] parent;
public UnionFind(int n) {
parent = new int[n];
for (int i = 0; i < n; ++i) {
parent[i] = i;
}
}
public int find(int p) {
while (parent[p] != p) {
parent[p] = parent[parent[p]];
p = parent[p];
}
return p;
}
public void union(int p, int q) {
int proot = find(p);
int qroot = find(q);
if (proot == qroot) return;
parent[proot] = qroot;
}
}
也可以使用两个map记录分别记录和
中每个连通分量中各个数字出现的次数,最后做比较计算结果。
class Solution {
public int minimumHammingDistance(int[] source, int[] target, int[][] allowedSwaps) {
int n = source.length;
UnionFind uf = new UnionFind(n);
for (int[] swap : allowedSwaps) {
uf.union(swap[0], swap[1]);
}
Map<Integer, Map<Integer, Integer>> s = new HashMap<>();
for (int i = 0; i < n; i++) {
Map<Integer, Integer> map = s.computeIfAbsent(uf.find(i), k -> new HashMap<>());
map.put(source[i], map.getOrDefault(source[i], 0) + 1);
}
Map<Integer, Map<Integer, Integer>> t = new HashMap<>();
for (int i = 0; i < n; i++) {
Map<Integer, Integer> map = t.computeIfAbsent(uf.find(i), k -> new HashMap<>());
map.put(target[i], map.getOrDefault(target[i], 0) + 1);
}
int ans = 0;
for (Map.Entry<Integer, Map<Integer, Integer>> entry : s.entrySet()) {
int deputy = entry.getKey();
Map<Integer, Integer> num2Cnt = entry.getValue();
for (Map.Entry<Integer, Integer> entry1 : num2Cnt.entrySet()) {
int num = entry1.getKey();
int cnt = entry1.getValue();
int diff = cnt - t.get(deputy).getOrDefault(num, 0);
if (diff > 0) {
ans += diff;
}
}
}
return ans;
}
}
class UnionFind {
int[] parent;
public UnionFind(int n) {
parent = new int[n];
for (int i = 0; i < n; ++i) {
parent[i] = i;
}
}
public int find(int p) {
while (parent[p] != p) {
parent[p] = parent[parent[p]];
p = parent[p];
}
return p;
}
public void union(int p, int q) {
int proot = find(p);
int qroot = find(q);
if (proot == qroot) return;
parent[proot] = qroot;
}
}