检查Python中矩阵中第i行和第i列的总和是否相同
假设我们有一个二维矩阵。我们必须检查第i行的总和是否与第i列的总和相同。
所以，如果输入是这样的
那么输出将为True，因为第一行和第一列的总和是(2+3+4+5)=14和(2+10+1+1)=14。
为了解决这个问题，我们将按照以下步骤操作-
row:=垫子的行数
col:=垫子的列数
total_row:=0,total_col:=0
对于范围0到第-1行的i，请执行
返回真
total_row:=total_row+mat[i,j]
total_col:=total_col+mat[j,i]
total_row:=0,total_col:=0
对于0到col-1范围内的j，执行
如果total_row与total_col相同，则
返回错误
让我们看看以下实现以获得更好的理解-
示例代码
def solve(mat):
row = len(mat)
col = len(mat[0])
total_row = 0
total_col = 0
for i in range(row):
total_row = 0
total_col = 0
for j in range(col):
total_row += mat[i][j]
total_col += mat[j][i]
if total_row == total_col:
return True
return False
matrix = [
[2,3,4,5],
[10,6,4,2],
[1,4,6,7],
[1,5,6,7]
]
print(solve(matrix))输入
[ [1,2,3,4], [9,5,3,1], [0,3,5,6], [0,4,5,6]]输出结果
True