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;
    }

 

posted @ 2021-09-19 14:52  旺仔古李  阅读(32)  评论(0编辑  收藏  举报