检查Python中矩阵中第i行和第i列的总和是否相同

• Z时代
• 2024-01-10
• 分类：综合

假设我们有一个二维矩阵。我们必须检查第 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

以上是 检查Python中矩阵中第i行和第i列的总和是否相同 的全部内容， 来源链接： utcz.com/z/345865.html