python解法：leetcode 542 0-1矩阵

class Solution(object):
    def updateMatrix(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[List[int]]
        """
        # BFS宽度优先搜索
        # 先找到所有的0点
        # 从0层开始搜索，与0相连的非0值全部为1
        # 继续搜索1层，与1相连的、未被遍历过的点且非0的点的值置为当前层1+1
        
        if not matrix:
            return 0
        m = len(matrix)
        n = len(matrix[0])
        visited = [[0]*n for i in range(m)]
        Q = []
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == 0:
                    visited[i][j] = 1
                    Q.append([i,j])
        dx = [-1,1,0,0]
        dy = [0,0,-1,1]
        cur_num = 0 
        while Q:
            size = len(Q)
            for _ in range(size):
                x, y = Q.pop(0)
                for i in range(4):
                    newx, newy = x + dx