2021-02-14

2021-02-14 Leetcode每日刷题

题目

N couples sit in 2N seats arranged in a row and want to hold hands. We want to know the minimum number of swaps so that every couple is sitting side by side. A swap consists of choosing any two people, then they stand up and switch seats.

The people and seats are represented by an integer from 0 to 2N-1, the couples are numbered in order, the first couple being (0, 1), the second couple being (2, 3), and so on with the last couple being (2N-2, 2N-1).
The couples’ initial seating is given by row[i] being the value of the person who is initially sitting in the i-th seat.

Example 1:

Input: row = [0, 2, 1, 3]
Output: 1
Explanation: We only need to swap the second (row[1]) and third (row[2]) person.

Example 2:

Input: row = [3, 2, 0, 1]
Output: 0
Explanation: All couples are already seated side by side.

Note:

len(row) is even and in the range of [4, 60].
row is guaranteed to be a permutation of 0…len(row)-1.

我的思路
是看了解答的贪心策略才敢写的。对于每组数字(2i, 2i+1)，当第一个数为偶数的时候检查第二个数是否比它大1，如果不是则搜索到之后交换一次。当第一个数为奇数的时候检查第二个数是否比它小1并进行交换。我也不知道这样为什么是对的，但是它的答案确实是对的。

class Solution:
    def minSwapsCouples(self, row: List[int]) -> int:
        cnt = 0
        for i in range(int(len(row)/2)):
            if row[2*i]%2 == 0:
                if row[2*i+1] != row[2*i] +1:
                    for j in range(2*i+1,len(row)):
                        if row[j] == row[2*i]+1:
                            row[2*i+1],row[j] = row[j],row[2*i+1]
                            cnt+=1 
            else:
                if row[2*i+1] != row[2*i] - 1:
                    for j in range(2*i+1,len(row)):
                        if row[j] == row[2*i]-1:
                            row[2*i+1],row[j] = row[j],row[2*i+1]
                            cnt+=1
        return cnt

提交结果

参考思路
关于为什么贪心解法是正确的：

• 从图的角度来讲，将N对情侣看做图中的 N 个节点；对于每对相邻的位置，如果是第i对与第j对坐在了一起，则在i号节点与j号节点之间连接一条边，代表需要交换这两对情侣的位置。

• 因为这题情侣是2人一对，同一连通分量内，每个点最多连出两条边，且必成环（可以用反证法；当连通分量内只有一个点时，说明该情侣坐对了，不需要换位置；只有两个点时，图中只有一条边，直接调换一次即可；在一般情况下，即连通分量内点数大于2时，此时每个点必须连出两条边[如果只连一条边的话，一定属于第二种只有两个点的情况；连三条边不可能，因为情侣是2人]！这一条件当且仅当成环的时候成立，反之会出现某些点只有一条边或者多余两条边的情况）。

• 至此，对于每个连通分量，由于其一定成环，所以我们直接沿着环的方向依次交换（边数-1）次即可使得该连通分量内情侣牵手，也就是贪心交换。

一个简化版的代码：

class Solution(object):
    def minSwapsCouples(self, row):
        """
        :type row: List[int]
        :rtype: int
        """
        N = len(row)
        res = 0
        for i in range(0, N - 1, 2):
            if row[i] == row[i + 1] ^ 1:
                continue
            for j in range(i + 1, N):
                if row[i] == row[j] ^ 1:
                    row[i + 1], row[j] = row[j], row[i + 1]
            res += 1
        return res

这里有个小技巧合并奇数偶数两种情况：

求数字 x 的对象时用到了一个技巧，x 的对象是x ^ 1。解释如下：

• 当 x 是偶数，则其二进制的末尾是 0，所以 x ^ 1 将其二进制的末尾改成 1，于是得到了x的对象 x + 1。 当 x
• 是奇数，则其二进制的末尾是 1，所以 x ^ 1 将其二进制的末尾改成 0，于是得到了x的对象 x - 1。

情人节算这个好心累。祝天下有情人终成眷属。
正好一周了，从明天开始加大力度多刷一道简单题。