检查矩阵是否可以通过在 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