leetcode 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 的一个全排列。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/couples-holding-hands
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分组递归查找
要是有两对情侣互相换了座位,则需要交换一次。
要是有三对情侣互相换了座位,则需要交换二次。
要是有四对情侣互相换了座位,则需要交换三次。
所以可得 n 对情侣换座位,需要 n - 1 次。
接下来就是把数组中的情侣,分成若干组,每一组中的成员都是需要相互交换座位的。
需要交换的总次数 = 每一组需要交换次数的和。
创建数组 arr 来记录每一个人的位置,方便之后的查找用。
递归查找分组,每查找一个,就把值设置为-1,防止重复查找。
private int[] arr; private int[] row; public int minSwapsCouples(int[] row) { int length = row.length; int[] arr = new int[length]; for (int i = 0; i < length; i++) { arr[row[i]] = i; } this.arr = arr; this.row = row; int sum = 0; for (int i = 0; i < length; i += 2) { if (row[i] == -1) { continue; } int count = find(row[i], row[i + 1]); row[i] = -1; row[i + 1] = -1; sum += count - 1; } return sum; } private int find(int a, int b) { int c = (a & 1) == 0 ? a + 1 : a - 1; int size = 0; if (c != b) { int i = arr[c]; int j = (i & 1) == 0 ? i + 1 : i - 1; int v = row[j]; row[j] = -1; row[i] = -1; size = find(v, b); } return size + 1; }