【HIHOCODER 1478】 水陆距离（BFS）

描述

---

给定一个N x M的01矩阵，其中1表示陆地，0表示水域。对于每一个位置，求出它距离最近的水域的距离是多少。
矩阵中每个位置与它上下左右相邻的格子距离为1。

输入

---

第一行包含两个整数，N和M。
以下N行每行M个0或者1，代表地图。
数据保证至少有1块水域。
对于30%的数据，1 <= N, M <= 100
对于100%的数据，1 <= N, M <= 800

输出

---

输出N行，每行M个空格分隔的整数。每个整数表示该位置距离最近的水域的距离。

样例输入

4 4  
0110  
1111  
1111  
0110

样例输出

0 1 1 0  
1 2 2 1  
1 2 2 1  
0 1 1 0

题解

---

以水域为起点，使用bfs搜索，一旦一个水域搜到一个陆地，其它水域就不必搜到这个陆地，因为这已经是最短路了

import java.util.*;
import java.io.*;
public class Main {
    static final int N = 810;
    static final int M = (int) 7e5 + 10;
    static final int inf = 0x3f3f3f3f;
    int[][] ans = new int[N][N];
    char[][] a = new char[N][N];
    int[][] book = new int[N][N];
    int n, m;
    class node {
        public int x, y, step;
        node() {
        };
        node(int tx, int ty, int s) {
            x = tx;
            y = ty;
            step = s;
        };
    }

    int nex[][] = { { 0, 1 }, { 1, 0 }, { -1, 0 }, { 0, -1 } };
    Queue<node> q = new LinkedList<>();

    void bfs() {
        node u = new node(), v = new node();
        while (!q.isEmpty()) {
            u = q.poll();
            if (a[u.x][u.y] == '1')
                ans[u.x][u.y] = Math.min(ans[u.x][u.y], u.step);
            for (int i = 0; i < 4; i++) {
                v.x = u.x + nex[i][0];
                v.y = u.y + nex[i][1];
                if (v.x >= n || v.y >= m || v.x < 0 || v.y < 0)
                    continue;
                if (book[v.x][v.y] == 1)
                    continue;
                book[v.x][v.y] = 1;
                v.step = u.step + 1;
                q.offer(new node(v.x, v.y, v.step));
            }
        }
    }

    void solve() {
        Scanner sc = new Scanner(new InputStreamReader(System.in));
        PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
        n = sc.nextInt();
        m = sc.nextInt();
        for (int i = 0; i < n; i++)
            Arrays.fill(ans[i], inf);
        for (int i = 0; i < n; i++) {
            a[i] = sc.next().toCharArray();
            for (int j = 0; j < m; j++) {
                if (a[i][j] == '0') {
                    q.offer(new node(i, j, 0));
                    ans[i][j] = 0;
                }
            }
        }
        bfs();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (j > 0)
                    out.print(" ");
                out.print(ans[i][j]);
            }
            out.println();
        }
        out.flush();
    }
    public static void main(String[] args) {
        Main a = new Main();
        a.solve();
    }
}

转载于:https://www.cnblogs.com/zsyacm666666/p/7656397.html