Pythonでの騎士の攻撃

長方形のチェス盤を表す2次元のバイナリ行列があるとします。ここで、0は空のセルを表し、1は騎士を表します。騎士は、水平方向に2マス、垂直方向に1マス、または縦に2マス、横に1マス移動することができます（チェス盤の騎士のように）。 2人の騎士がお互いを攻撃しているかどうかを確認する必要があります。

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

その後、出力はTrueになります

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

• row、col：=行列の行数、行列の列数
• 0から行1の範囲のrの場合、実行
  • 0からcol-1の範囲のcの場合、do
    • A [r] [c]がゼロ以外の場合、
      • 各nr、nc in [（r + 1、c-2）、（r + 1、c + 2）、（r + 2、c-1）、（r + 2、c + 1）] 、do
        • nrが行の範囲内にあり、ncがcolの範囲内にあり、A [nr、nc]がゼロ以外の場合、
          • Trueを返す
• Falseを返す

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

例

class Solution:
   def solve(self, A):
      row, col = len(A), len(A[0])
      for r in range(row):
         for c in range(col):
            if A[r][c]:
               for nr, nc in ((r+1, c-2), (r+1, c+2), (r+2, c-1), (r+2, c+1)):
                  if 0 <= nr < row and 0 <= nc <col and
                     A[nr][nc]:
                     return True
      return False
ob = Solution()
mat = [[0,0,0,0,0],
[0,1,0,0,0],
[0,0,0,1,0]]
print(ob.solve(mat))

入力

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

出力

True