Python 中計算訪問城市街區的總距離


假設我們有一個由唯一字串表示的城市街區的矩陣,以及另一個包含要訪問的街區的字串列表。如果我們位於街區 matrix[0][0],則查詢按順序訪問每個街區所需的總曼哈頓距離。

因此,如果輸入類似於

QBC
DEZ
GGi

Block = [H,B,C]

那麼輸出將為 6,因為“h”在下方(南)2 個街區,右側(東)1 個街區,“b”在上方(北)2 個街區,“c”在右側(東)1 個街區。

要解決此問題,我們將遵循以下步驟:

  • coords := 一個鍵為 'start',值為 (0, 0) 的對映
  • 對於 mat 中的每一行,執行
    • 對於 mat 中的每一列,執行
      • 將 (row,col) 插入到 coords[mat[row, col]] 中
  • dist := 0
  • 更新 blocks,在開頭新增 'start'
  • 對於 i 在 0 到 blocks 大小 -1 的範圍內,執行
    • c1 := coords[blocks[i]]
    • c2 := coords[blocks[i+1]]
    • d := |c1[0]-c2[0]| + |c1[1]-c2[1] |
    • dist := dist + d
  • 返回 dist

讓我們看看以下實現以更好地理解:

示例

 即時演示

class Solution:
   def solve(self, mat, blocks):
      coords = {'start': (0,0)}
      for row in range(len(mat)):
         for col in range(len(mat[row])):
            coords[mat[row][col]] = (row,col)
      dist = 0
      blocks = ['start']+blocks
      for i in range(len(blocks)-1):
         c1 = coords[blocks[i]]
         c2 = coords[blocks[i+1]]
         d = abs(c1[0]-c2[0]) + abs(c1[1]-c2[1])
         dist += d
      return dist
ob = Solution()
inp = [["q", "b", "c"],
["d", "e", "z"],
["g", "h", "i"]]
blk = ["h", "b", "c"]
print(ob.solve(inp, blk))

輸入

[["q", "b", "c"],["d", "e", "z"],["g", "h", "i"]]

輸出

6

更新於: 2020-09-22

175 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

立即開始
廣告

© . All rights reserved.