1. 问题描述:
给你一个 n x n 的二进制网格 grid,每一次操作中,你可以选择网格的相邻两行进行交换。一个符合要求的网格需要满足主对角线以上的格子全部都是 0 。请你返回使网格满足要求的最少操作次数,如果无法使网格符合要求,请你返回 -1 。主对角线指的是从 (1, 1) 到 (n, n) 的这些格子。
示例 1:
输入:grid = [[0,0,1],[1,1,0],[1,0,0]]
输出:3
示例 2:
输入:grid = [[0,1,1,0],[0,1,1,0],[0,1,1,0],[0,1,1,0]]
输出:-1
解释:所有行都是一样的,交换相邻行无法使网格符合要求。
示例 3:
输入:grid = [[1,0,0],[1,1,0],[1,1,1]]
输出:0
2. 思路分析:
我们可以使用一个循环统计出每一行中右边连续的0的个数,并且使用一个列表来记录对应的结果,这样才2我们就可以知道当前这一行的右边连续0的个数,然后就可以将这一行需要的0的个数对应的行数交换到这一行中,然后我们可以通过下标从0到方阵的倒数第二行,计算出当前遍历下标i对应的行数需要连续的0的个数,然后从当前这一行往下寻找到需要连续0的个数对应的行数,而找到的这一行与当前的行数的差值就是找到这一行对应的连续的0的行数交换到当前行需要交换的次数(从下往上进行交换),因为交换了相邻的两行,所以需要使用循环来修改记录连续的0的个数的列表中对应的相邻两行的值
3. 代码如下:
from typing import List
class Solution:
# 注意是方阵
def minSwaps(self, grid: List[List[int]]) -> int:
length = len(grid)
# 记录每一行中0的个数
zeros = list()
for i in range(length):
j = length - 1
while j >= 0 and grid[i][j] == 0: j -= 1
zeros.append(length - 1 - j)
res = 0
# 到倒数第二行即可
for i in range(length - 1):
needzeros = length - 1 - i
j = i
while j < length and zeros[j] < needzeros: j += 1
# 没有找到当前这一行需要的0的个数
if j == length: return -1
res += j - i
# 将当前这一行需要的0的个数对应的行通过相邻行交换到这一行
while j > i:
zeros[j], zeros[j - 1] = zeros[j - 1], zeros[j]
j -= 1
return res