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

假设我们有一个二维矩阵。我们必须检查第 i 行的总和是否与第 i 列的总和相同。

因此，如果输入类似于

2
3
4
5
10
6
4
2
1
4
6
7
1
5
6
7

则输出将为 True，因为第一行和第一列的总和为 (2 + 3 + 4 + 5) = 14，而 (2 + 10 + 1 + 1) = 14。

为了解决这个问题，我们将遵循以下步骤：

  • row := mat 的行数
  • col := mat 的列数
  • total_row := 0, total_col := 0
  • 对于 i 的范围从 0 到 row - 1，执行
    • total_row := 0, total_col := 0
    • 对于 j 的范围从 0 到 col - 1，执行
      • total_row := total_row + mat[i, j]
      • total_col := total_col + mat[j, i]
    • 如果 total_row 与 total_col 相同，则
      • 返回 True
  • 返回 False

让我们看看下面的实现，以便更好地理解：

示例代码

在线演示

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

更新于： 2021年1月15日

197 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告