检查矩阵是否可以通过在 Python 中转置方形子矩阵来转换为另一个矩阵
假设我们有两个NXM,分别称为mat1和mat2。在一个操作中,我们可以转置mat1中的任何方形子矩阵。我们必须通过执行给定的操作来检查是否可以从mat1获取mat2。
所以,如果输入是这样的
那么输出将为True,因为如果我们得到mat1的大小为2x2的右上角子矩阵的转置,我们将得到mat2。
示例
让我们看看以下实现以获得更好的理解-
def solve(mat1, mat2): row = len(mat1) column = len(mat1[0]) for i in range(row): temp1 = [] temp2 = [] r = i col = 0 while r >= 0 and col < column: temp1.append(mat1[r][col]) temp2.append(mat2[r][col]) r -= 1 col += 1 temp1.sort() temp2.sort() for i in range(len(temp1)): if temp1[i] != temp2[i]: return False for j in range(1, column): temp1 = [] temp2 = [] r = row - 1 col = j while r >= 0 and col < column: temp1.append(mat1[r][col]) temp2.append(mat2[r][col]) r -= 1 col += 1 temp1.sort() temp2.sort() for i in range(len(temp1)): if temp1[i] != temp2[i]: return False return True mat1 = [ [5, 6, 7], [1, 2, 3], [6, 8, 9]] mat2 = [ [5, 6, 2], [1, 7, 3], [6, 8, 9]] print(solve(mat1, mat2))
输入
[ [5, 6, 7], [1, 2, 3], [6, 8, 9]], [ [5, 6, 2], [1, 7, 3], [6, 8, 9]]输出结果
True