Java实现 LeetCode 765 情侣牵手(并查集 || 暴力)
765. 情侣牵手
N 对情侣坐在连续排列的 2N 个座位上,想要牵到对方的手。 计算最少交换座位的次数,以便每对情侣可以并肩坐在一起。 一次交换可选择任意两人,让他们站起来交换座位。
人和座位用 0 到 2N-1 的整数表示,情侣们按顺序编号,第一对是 (0, 1),第二对是 (2, 3),以此类推,最后一对是 (2N-2, 2N-1)。
这些情侣的初始座位 row[i] 是由最初始坐在第 i 个座位上的人决定的。
示例 1:
输入: row = [0, 2, 1, 3]
输出: 1
解释: 我们只需要交换row[1]和row[2]的位置即可。
示例 2:
输入: row = [3, 2, 0, 1]
输出: 0
解释: 无需交换座位,所有的情侣都已经可以手牵手了。
说明:
len(row) 是偶数且数值在 [4, 60]范围内。
可以保证row 是序列 0…len(row)-1 的一个全排列。
class Solution {
private int[] parent;
public int minSwapsCouples(int[] row) {
int n = row.length;
parent = new int[n];
for (int i = 0; i < n; i += 2) {
parent[i] = i;
parent[i + 1] = i;
}
for (int i = 0; i < n; i += 2) {
union(row[i], row[i + 1]);
}
int res = n / 2;
for (int i = 0; i < n; i++) {
if (parent[i] == i) {
res--;
}
}
return res;
}
private int find(int node) {
return parent[node] == node ? node : (parent[node] = find(parent[node]));
}
private void union(int node1, int node2) {
int root1 = find(node1);
int root2 = find(node2);
if (root1 == root2) {
return;
}
parent[root1] = root2;
}
}
class Solution {
public int minSwapsCouples(int[] row) {
int res = 0;
for (int i = 0; i < row.length; i += 2) {
int t = row[i] ^ 1;
if (row[i + 1] == t) {
continue;
}
res++;
for (int j = i + 1; j < row.length; j++) {
if (row[j] == t) {
// 交换
row[j] = row[i + 1];
row[i + 1] = t;
break;
}
}
}
return res;
}
}