542. 01 矩阵

0-1矩阵，就是给定一个01矩阵，得到另一个矩阵，该矩阵表示每个元素距离0最近的距离。
（1）
0 0 0
0 1 0
0 0 0
输出
0 0 0
0 1 0
0 0 0
（2）
0 0 0
0 1 0
1 1 1
输出
0 0 0
0 1 0
1 2 1

寻找最短距离，就是BFS算法

from collections import deque
class Solution:
    directions=[(-1,0),(0,-1),(1,0),(0,1)]
    def updateMatrix(self,matrix):
        row=len(matrix)
        col=len(matrix[0])
        res=[[-1 for _ in range(col)]for _ in range(row)] #初始化-1，表示没有访问。不像岛屿初始化为False,是因为后面以res[][]的值做+1.
        queue=deque()
        for i in range(row):
            for j in range(col):
                if matrix[i][j]==0: # 0的矩阵里的最短距离还是0
                    queue.append((i,j))
                    res[i][j]=0
        while queue:
            cur_i,cur_j=queue.popleft()
            for direction in self.directions:
                new_i=cur_i+direction[0]
                new_j=cur_j+direction[1]
                if 0<=new_i<row and 0<=new_j<col and res[new_i][new_j]== -1:
                    res[new_i][new_j]=res[cur_i][cur_j]+1 #表明queue.pop出来的数据的上下四周有个元素符合res[new_i][new_j]== -1；那么就让原点+1.
                    queue.append((new_i,new_j))#也许下一个1的上下左右还是个1，而这个1是前期从0开始append过的。所以append[new_i][new_j]
        return res