行列内の要素の数を見つけるプログラムは、Pythonの行列の基準に従います

バイナリ行列があるとします。次のルールに従う行列内の要素の数を見つける必要があります-

• matrix [r、c] =1

• jがcと等しくない場合はすべてのjに対してmatrix[r、j] =0であり、iがrと等しくない場合はすべてのiに対してmatrix [i、c]=0です。

したがって、入力が次のような場合

基準を満たすセル（0,2）、（1,0）、（2,1）があるため、出力は3になります。

これを解決するには、次の手順に従います-

• 行列が空の場合、

  • 0を返す

• row：=マトリックス内のすべての行エントリの合計のリスト

• col：=マトリックス内のすべての列エントリの合計のリスト

• m：=行列の行数

• n：=行列の列数

• res：=0

• 0からm-1の範囲のrの場合、実行

  • 0からn-1の範囲のcの場合、実行

    • matrix [r、c]が1で、row [r]が1で、col [c]も1の場合、

      • res：=res + 1

• 解像度を返す

例

理解を深めるために、次の実装を見てみましょう

def solve(matrix):
   if not matrix:
      return 0

   row = [sum(r) for r in matrix]
   col = [sum(c) for c in zip(*matrix)]

   m, n = len(matrix), len(matrix[0])
   res = 0
   for r in range(m):
      for c in range(n):
         if matrix[r][c] == 1 and row[r] == 1 and col[c] == 1:
            res += 1
   return res

matrix = [
   [0, 0, 1],
   [1, 0, 0],
   [0, 1, 0]
]
print(solve(matrix))

入力

[[0, 0, 1],[1, 0, 0],[0, 1, 0]]

出力

3