题目
N 对情侣坐在连续排列的 2N 个座位上,想要牵到对方的手。 计算最少交换座位的次数,以便每对情侣可以并肩坐在一起。 一次交换可选择任意两人,让他们站起来交换座位。
人和座位用 0 到 2N-1 的整数表示,情侣们按顺序编号,第一对是 (0, 1),第二对是 (2, 3),以此类推,最后一对是 (2N-2, 2N-1)。
这些情侣的初始座位 row[i] 是由最初始坐在第 i 个座位上的人决定的。
示例
输入: row = [0, 2, 1, 3]
输出: 1
解释: 我们只需要交换row[1]和row[2]的位置即可。
说明
- len(row) 是偶数且数值在 [4, 60]范围内。
- 可以保证row 是序列 0…len(row)-1 的一个全排列
题解
- 假如有N对情侣,对于相邻的元素,若该相邻元素是一对情侣,我们将其相连;若不是一对情侣,我们也将其相连
- 是一对情侣: 0, 1, 2, 3,交换次数为0;
- 不是一对情侣:1, 3, 0, 2,此时要么交换1与2,或者交换0和3
- 对于第N对情侣,其值中较小的数为偶数且奇数=偶数+1。其特点是除以2得到的数相等
- 无论是坐对位置或者是坐错位置逻辑上连在一起的情侣的对数N1, N2, N3…Nn
(n为连通分量的个数)
则N1+N2+n3+…+Nn=N,然而把这些连在一起的情侣拆开,至少需要交换N1-1,N2-1,N3-1…Nn-1次 - 即:N1+N2+N3+…+Nn-n=N-n。「至少交换的次数 = 所有情侣的对数 - 并查集里连通分量的个数」。
参考代码
class Solution {
public int minSwapsCouples(int[] row) {
UnionFind myUnionFind = new UnionFind(row.length/2);
for(int i = 0; i < row.length; i += 2) {
int a = row[i]/2;
int b = row[i+1]/2;
myUnionFind.union(a, b);
}
return row.length/2 - myUnionFind.getCount();
}
class UnionFind {
int[] parent;
int[] rank;
int count;
public UnionFind(int n) {
this.count = n;
parent = new int[n];
rank = new int[n];
Arrays.fill(rank, 1);
for(int i = 0; i < n; i++) {
parent[i] = i;
}
}
public int getCount() {
return count;
}
public void union(int a, int b) {
int rootA = find(a);
int rootB = find(b);
if(rootA == rootB) return;
if(rank[rootA] < rank[rootB]) {
parent[rootA] = rootB;
} else {
if(rank[rootA] == rank[rootB]) {
rank[rootA]++;
}
parent[rootB] = rootA;
}
count--;
}
public int find(int index) {
if(index != parent[index]) {
return find(parent[index]);
}
return index;
}
}
}