173. 矩阵距离
摘要
Title: 173. 矩阵距离
Tag: 多源BFS
Memory Limit: 64 MB
Time Limit: 1000 ms
Powered by:NEFU AB-IN
173. 矩阵距离
-
题意
求矩阵中所有点到距离自己最近的“1”点的最短距离是多少
-
思路
总体来说,就是将所有能走的地点,全部放进队列,并打上标记,其余的就和单源BFS相同了 -
代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42'''
Author: NEFU AB-IN
Date: 2022-04-12 16:29:20
FilePath: \ACM\Acwing\173.py
LastEditTime: 2022-04-12 16:34:34
'''
N = 1010
from collections import deque
g = []
dist = [[0] * N for _ in range(N)]
st = [[0] * N for _ in range(N)]
n, m = map(int, input().split())
for i in range(n):
g.append(list(map(int, input())))
q = deque()
for i in range(n):
for j in range(m):
if g[i][j] == 1:
st[i][j] = 1
q.appendleft([i, j])
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
while q:
x, y = q.pop()
for i in range(4):
a = x + dx[i]
b = y + dy[i]
if a >= 0 and a < n and b >= 0 and b < m and st[a][b] == 0:
dist[a][b] = dist[x][y] + 1
st[a][b] = 1
q.appendleft([a, b])
for i in range(n):
for j in range(m):
print(dist[i][j], end=" ")
print()