給定一個由 0 和 1 組成的矩陣 mat，請輸出一個大小相同的矩陣，其中每一個格子是 mat 中對應位置元素到最近的 0 的距離。

兩個相鄰元素間的距離為 1 。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/01-matrix
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

多源廣度優先搜尋

import java.util.LinkedList;
import java.util.Queue;

class Solution {

    private int[] dx = {0, 1, 0, -1};

    private int[] dy = {1, 0, -1, 0};

    public int[][] updateMatrix(int[][] mat) {
        if (mat == null || mat.length == 0 || mat[0].length == 0) {
            return new int[0][0];
        }

        int n = mat.length, m = mat[0].length;
        int[][] ret = new int[n][m];
        Queue<Operation> queue = new LinkedList<>();
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if (mat[i][j] == 0) {
                    queue.offer(new Operation(i, j, 0));
                }
            }
        }

        while (!queue.isEmpty()) {
            Operation operation = queue.poll();
            ret[operation.x][operation.y] = operation.step;
            for (int i = 0; i < 4; ++i) {
                int x = dx[i] + operation.x;
                int y = dy[i] + operation.y;
                if (x >= 0 && x < n && y >= 0 && y < m && mat[x][y] == 1) {
                    queue.offer(new Operation(x, y, operation.step + 1));
                    mat[x][y] = 0;
                }
            }
        }
        return ret;
    }
}

class Operation {
    int x;
    int y;
    int step;

    public Operation(int x, int y, int step) {
        this.x = x;
        this.y = y;
        this.step = step;
    }
}
 

動態規劃