AcWing 173. 矩阵距离

AcWing 173. 矩阵距离

题目链接：

AcWing 173. 矩阵距离

正经题意：

给定一个 $N$ 行 $M$ 列的 01矩阵 $A$，$A[i][j]$与 $A[k][l]$ 之间的曼哈顿距离定义为：
$dist(A[i][j],A[k][l])=|i-k|+|j-l|$

输出一个 $N$行 $M$列的整数矩阵 $B$，其中：
$B[i][j]=mi{n}_{1\le x\le N,1\le y\le M,A[x][y]=1}dist(A[i][j],A[x][y])$

简而言之：

就是这上面的每个点到1的最短距离

思路：

如果只有一个1，那就是直接入队bfs，那么有多个一是否也是如此，答案是肯定的
bfs的性质是第一次遇见一定是最短路，所以所有的1入队，bfs遍历的所有格子都是最短的。也就是多源bfs

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
typedef pair<int,int> PII;
#define x first
#define y second
int n,m;
int g[N][N];
int st[N][N];//状态
int d[N][N];//距离
queue<PII> q;
int dx[] = {0,0,1,-1};
int dy[] = {1,-1,0,0};

int bfs(){
    while(q.size()){
        auto t = q.front();
        q.pop();
        for(int i = 0;i<4;i++){
            int cx = t.x + dx[i];
            int cy = t.y + dy[i];
            if(cx > n|| cy > m|| cx < 1||cy < 1) continue;
            if(st[cx][cy]) continue;
            st[cx][cy] = 1;
            d[cx][cy] = d[t.x][t.y] + 1;
            q.push({cx,cy});
        }        
    }
}
int main(){
    cin>>n>>m;
    for(int i = 1;i<=n;i++){
        for(int j = 1;j<=m;j++){
            scanf("%1d",&g[i][j]);
            if(g[i][j] == 1){
                q.push({i,j});
                st[i][j] = 1;
                d[i][j] = 0;
            }
        }
    }
    bfs();
    for(int i = 1;i<=n;i++){
        for(int j = 1;j<=m;j++){
            cout<<d[i][j]<<" ";
        }
        cout<<endl;
    }
}