Leetcode 765.情侣牵手

Posted on |

765. 情侣牵手

题目

N 对情侣坐在连续排列的 2N 个座位上，想要牵到对方的手。 计算最少交换座位的次数，以便每对情侣可以并肩坐在一起。 一次交换可选择任意两人，让他们站起来交换座位。

人和座位用 0 到 2N-1 的整数表示，情侣们按顺序编号，第一对是 (0, 1)，第二对是 (2, 3)，以此类推，最后一对是 (2N-2, 2N-1)。

这些情侣的初始座位 row[i] 是由最初始坐在第 i 个座位上的人决定的。

1
2
3
4
5
6
7
8
9
示例 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 的一个全排列。

方法

方法1：贪心算法

贪心，不过不好证明。从位置0开始，看位置1是否是它的partner，如果是，直接忽略这组couple，如果不是，从后面的数组中，找到partner直接交换，并且记录交换次数一次。遍历完整个数组即为最优解。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution(object):
    def minSwapsCouples(self, row):
        """
        :type row: List[int]
        :rtype: int
        """
        pos,step=0,0

        while pos<len(row):
            wife=row[pos]
            husband=row[pos+1]

            if wife%2==0:
                if wife+1==husband:
                    pass
                else:
                    husband_pos=row.index(wife+1)
                    row[pos+1],row[husband_pos]=row[husband_pos],row[pos+1]
                    step+=1
            else:
                if wife-1==husband:
                    pass
                else:
                    husband_pos=row.index(wife-1)
                    row[pos+1],row[husband_pos]=row[husband_pos],row[pos+1]
                    step+=1
            pos+=2
        return step
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution(object):
    def minSwapsCouples(self, row):
        """
        :type row: List[int]
        :rtype: int
        """
        s=[i//2 for i in row]
        ans=0

        for i in range(1,len(s)-2,2):
            if s[i]!=s[i-1]:
                couple=s.index(s[i-1],i)
                s[couple]=s[i]
                s[i]=s[i-1]
                ans+=1
        return ans

参考资料