Leetcode每日一题0214——765. 情侣牵手（Python）

题目内容

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 的一个全排列。

代码

# DFS方法 

class Solution:
    def minSwapsCouples(self, row: List[int]) -> int:
        n = len(row)
        tot = n//2  # 共有tot对情侣

		# graph[i]存放与第 i+1 对情侣坐在一起的其他情侣编号
        graph = [ [] for _ in range(tot)  ]

        for i in range(0, n, 2):
            l = row[i]//2      # row[i]位置上坐的人属于第几对情侣
            r = row[i+1]//2    # row[j]位置上坐的人属于第几对情侣
            if l != r:         # 若两个人不是一对
                graph[l].append(r)    # 记录第l对情侣与第r对坐在了一起
                graph[r].append(l)    # 记录第r对情侣与第l对坐在了一起

        vis = [ 0 for _ in range(tot)]  # 第i对情侣是否已经访问

        res = 0

        # 依次遍历各对情侣
        for i in range(0, tot):
            if vis[i] == 0:
                queue = []
                vis[i] = 1
                queue.append(i)
                cnt = 0

				# 对队列中的情侣对front进行调整，计数
				# 对与其坐错的情侣加入队列，循环调整
                while queue != []:
                    front = queue[0]
                    queue.remove(front)
                    cnt += 1
					
                    for nei in graph[front]:
                        if vis[nei] == 0:
                            vis[nei] = 1
                            queue.append(nei)
                res += cnt -1
        
        return res