Pythonのグリッドボックスのどこにボールが着地するかを見つけるプログラム

m x nグリッドボックスが与えられたとします。各セルには、右上から左下、または左上から右下のいずれかに配置されたボードがあります。上のセルからボールがボックスに入れられ、そのボールがボックスの下部に到達するかどうかを確認する必要があります。グリッドはマトリックスとして与えられます。セルに1のマークが付いている場合、対角線上のボードは左上から右下に広がります。 -1とマークされている場合は、右上隅から左下隅にまたがっています。 n個のボールが箱に入れられた場合、底に到達するボールの数を調べる必要があります。

3x3グリッドボックスの例。

したがって、入力がmat =

その場合、出力は[-1、-1、-1、-1]

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

• i：=マットの行数

• j：=マットの列数

• res：=新しいリスト

• 0からjの範囲のvalの場合、実行

  • x：=val

  • 0からiの範囲のrについては、次のようにします

    • s：=mat [r、x]

    • x：=x + mat [r、x]

    • x<0またはx>=jまたはmat[r、x]がsと同じでない場合、

      • resの最後に要素-1を追加します

      • ループから出てきます

  • それ以外の場合

    • resの最後にxを追加します

• 解像度を返す

例

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

def solve(mat):
   i, j = map(len, (mat, mat[0]))
   res = []
   for val in range(j):
      x = val
      for r in range(i):
         s = mat[r][x]
         x += mat[r][x]
         if x < 0 or x >= j or mat[r][x] != s:
            res += [-1]
            break
      else:
         res += [x]
   return res

print(solve([[1, 1, 1, -1], [-1, 1, 1, -1], [1, -1, -1, 1],[1, -1, 1, -1] ]))

入力

[[1, 1, 1, -1], [-1, 1, 1, -1], [1, -1, -1, 1],[1, -1, 1, -1] ]

出力

[-1, -1, -1, -1]