假设我们有一个二维矩阵,其中存在这些值:0表示一个空单元格。1代表墙。2代表一个人。现在,一个人可以沿上,下,左,右四个方向中的任一方向行走,否则将停留在一个时间单位内。我们必须找到一个可步行的牢房,以使每个人见面并返回时间所需的时间减至最少。我们必须记住,两个人可以走过同一个空单元,并且您可以假设任何两个人之间总有一条路。
所以,如果输入像
2 | 0 | 1 | 0 |
1 | 0 | 0 | 2 |
2 | 0 | 2 | 0 |
那么输出将为2,因为所有位置都可以在位置矩阵[1,1]处满足,且最多需要2步。
class Solution: def solve(self, A): R, C = len(A), len(A[0]) def get_neighbor(r, c): for nr, nc in ((r − 1, c), (r, c − 1), (r + 1, c), (r, c + 1)): if 0 <= nr < R and 0 <= nc < C and A[nr][nc] & 1 == 0: yield nr, nc def bfs(r, c): queue = [(r, c)] dist = {(r, c): 0} for r, c in queue: if dist[r, c] > 15: break for nr, nc in get_neighbor(r, c): if (nr, nc) not in dist: dist[nr, nc] = dist[r, c] + 1 queue.append((nr, nc)) return dist dist = None for r, row in enumerate(A): for c, val in enumerate(row): if val == 2: ndist = bfs(r, c) if dist is None: dist = ndist else: for key in list(dist.keys()): if key in ndist: dist[key] = max(dist[key],ndist[key]) else: del dist[key] return min(dist.values()) if dist else 0 ob = Solution() matrix = [ [2, 0, 1, 0], [1, 0, 0, 2], [2, 0, 2, 0] ] print(ob.solve(matrix))
[ [2, 0, 1, 0], [1, 0, 0, 2], [2, 0, 2, 0] ]输出结果
2