检查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
热门推荐
10 新人红包祝福语简短精辟
11 幼师给毕业祝福语简短
12 生日高考祝福语妹妹简短
13 下飞机祝福语简短英文
14 元宵发客户祝福语简短
15 爸妈金婚文案祝福语简短
16 新郎朋友红包祝福语简短
17 虎年五一祝福语大全简短
18 出阁宴祝福语姐姐简短