2501 矩阵距离 （bfs）

描述

给定一个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]=min(1≤x≤N,1≤y≤M,A[x][y]=1)⁡{dist(A[i][j],A[x][y])}
即求与每个位置曼哈顿距离最近的1
N,M≤1000。

输入格式

第一行两个整数n,m。

接下来一个N行M列的01矩阵，数字之间没有空格。

输出格式

一个N行M列的矩阵B，相邻两个整数之间用一个空格隔开。

样例输入

3 4
0001
0011
0110

样例输出

3 2 1 0
2 1 0 0
1 0 0 1


思路：直接对每个1的起点bfs，搜过的点就不用搜了，每个第一次遍历的点，就是其中一个起点到其的最小曼哈顿距离

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
vector<P>v;
int r,c;
char maps[1005][1005];
int vis[1005][1005];
int way[4][2] = {1,0,-1,0,0,1,0,-1};
struct Node
{
    int x,y;
};
int num;
void init()
{
    scanf("%d%d",&r,&c);
    char s[1005];
    for(int i=1;i<=r;i++)
    {
        scanf("%s",s);
        for(int j=1;j<=c;j++)
        {
            maps[i][j] = s[j-1];
            if(maps[i][j] == '1')
            {
                num++;
                v.push_back(P(i,j));
                vis[i][j] = 1;
            }
        }
    }
}

void bfs()
{
    queue<P>que;
    while(!que.empty())que.pop();
    for(int i=0;i<v.size();i++)que.push(v[i]);
    while(!que.empty())
    {
        P tmp = que.front();
        que.pop();
        for(int i=0;i<4;i++)
        {
            int xx = tmp.first + way[i][0];
            int yy = tmp.second + way[i][1];
            if(xx >= 1 && xx <= r && yy >= 1 && yy <= c && !vis[xx][yy])
            {
                num++;
                vis[xx][yy] = vis[tmp.first][tmp.second] + 1;
                que.push(P(xx,yy));
            }
        }
        if(num == r * c)return;
    }
}

void print()
{
    for(int i=1;i<=r;i++)
    {
        for(int j=1;j<=c;j++)
        {
            printf(j==c?"%d
":"%d ",vis[i][j]-1);
        }
    }
}
int main()
{
    init();
    bfs();
    print();
}