BankinPythonで警備員からの最短距離を見つける

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

• M：=5

• N：=5

• dir_row：=[-1、0、1、0]

• dir_col：=[0、1、0、-1]

• 関数is_ok（）を定義します。これにはi、jが必要です

• （iが範囲0およびM）または（jが範囲0およびj N）の場合、

  • Falseを返す

• Trueを返す

• 関数isSafe（）を定義します。これには、i、j、matrix、resultが必要です

• matrix [i、j]が'O'でない場合、またはresult [i、j]が-1でない場合、

  • Falseを返す

• Trueを返す

• 関数calculate_dist（）を定義します。これは行列を取ります

• 結果：=次数M x Nの行列を作成し、-1で埋めます

• 両端キューを定義するq

• 0からMの範囲のiの場合、実行

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

    • result [i、j]：=-1

    • matrix [i、j]が'G'と同じ場合、

      • pos：=[i、j、0]

      • qの左側にposを挿入します

      • result [i、j]：=0

• q> 0のサイズがゼロ以外の場合、実行

  • curr：=qから最後の要素を削除

  • x、y、dist：=curr [0]、curr [1]、curr [2]

  • 0から3の範囲のiの場合、実行

    • is_ok（x + dir_row [i]、y + dir_col [i]）がゼロ以外であり、isSafe（x + dir_row [i]、y + dir_col [i]、matrix、result）がゼロ以外の場合、

      • result [x + dir_row [i]、y + dir_col [i]]：=dist + 1

      • pos：=[x + dir_row [i]、y + dir_col [i]、dist + 1]

      • qの左側にposを挿入します

• 0からMの範囲のiの場合、実行

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

    • 結果の表示[i、j]

  • 次の行に移動

例

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

from collections import deque as queue
M = 5
N = 5
dir_row = [-1, 0, 1, 0]
dir_col = [0, 1, 0, -1]
def is_ok(i, j):
   if ((i < 0 or i > M - 1) or (j < 0 or j > N - 1)):
      return False
   return True
def isSafe(i, j,matrix, result):
   if (matrix[i][j] != 'O' or result[i][j] != -1):
      return False
   return True
def calculate_dist(matrix):
   result = [[ -1 for i in range(N)]for i in range(M)]
   q = queue()
   for i in range(M):
      for j in range(N):
         result[i][j] = -1
         if (matrix[i][j] == 'G'):
            pos = [i, j, 0]
            q.appendleft(pos)
            result[i][j] = 0
   while (len(q) > 0):
      curr = q.pop()
      x, y, dist = curr[0], curr[1], curr[2]
   for i in range(4):
      if is_ok(x + dir_row[i], y + dir_col[i]) and isSafe(x + dir_row[i], y + dir_col[i], matrix, result) : result[x + dir_row[i]][y + dir_col[i]] = dist + 1
      pos = [x + dir_row[i], y + dir_col[i], dist + 1]
      q.appendleft(pos)
   for i in range(M):
      for j in range(N):
         if result[i][j] > 0:
            print(result[i][j], end=" ")
         else:
            print(result[i][j],end=" ")
      print()

matrix = [['O', 'O', 'O', 'O', 'G'],
   ['O', 'O', 'O', 'W', 'O'],
   ['O', 'W', 'O', 'O', 'O'],
   ['G', 'W', 'W', 'W', 'O'],
   ['O', 'O', 'O', 'O', 'G']]

calculate_dist(matrix)

入力

[['O', 'O', 'O', 'O', 'G'],
['O', 'O', 'O', 'W', 'O'],
['O', 'W', 'O', 'O', 'O'],
['G', 'W', 'W', 'W', 'O'],
['O', 'O', 'O', 'O', 'G']]

出力

3 3 2 1 0
2 3 3 -1 1
1 -1 4 3 2
0 -1 -1 -1 1
1 2 2 1 0