力扣(LeetCode)765情侣牵手

最新推荐文章于 2022-05-23 23:58:46 发布

原创最新推荐文章于 2022-05-23 23:58:46 发布 · 309 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

22 篇文章

订阅专栏

这是一个关于算法的问题，具体是情侣牵手的优化问题。给定一个初始排列，需要计算最少交换座位的次数，使得每对情侣可以并肩坐在一起。通过实现并查集数据结构，可以有效地找到每对情侣的连接关系，并计算所需的交换次数。示例中展示了如何通过并查集的find和join方法解决这个问题。

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