765. 情侣牵手
恰逢2.14情人节,力扣每日一题出了个这种题。让我这单身狗来会会这个题
问题
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
我的解答
并查集
class Solution:
def __init__(self):
self.dic = {}
self.count = 0
def find(self, value):
if value not in self.dic.keys():
self.dic[value] = value
self.count += 1
if self.dic[value] != value:
self.dic[value] = self.find(self.dic[value])
return self.dic[value]
def join(self, value1, value2):
root1 = self.find(value1)
root2 = self.find(value2)
if root1 != root2:
self.dic[root1] = root2
self.count -= 1
def double(self, value):
"""偶数化"""
if value % 2 == 1:
return value - 1
else:
return value
def minSwapsCouples(self, row: List[int]) -> int:
for i in range(0, len(row), 2):
value1 = self.double(row[i])
value2 = self.double(row[i + 1])
self.join(value1, value2)
return len(row) // 2 - self.count