矩阵距离

题目描述

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

#include<bits/stdc++.h>
using namespace std;

const int N=1010;
const int dx[4]={0,0,-1,1};
const int dy[4]={-1,1,0,0};

struct node{
	int x,y;
	node(){}
	node(int x,int y):x(x),y(y){}
};
queue<node>q;
int d[N][N];
char s[N][N];
int n,m;

void bfs(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			if(s[i][j]=='1'){
				q.push(node(i,j));
				d[i][j]=0;
			}
		}
	}
	while(!q.empty()){
		node now=q.front();q.pop();
		for(int i=0;i<4;i++){
			node nxt=now;
			nxt.x+=dx[i];nxt.y+=dy[i];
			if(nxt.x>=1&&nxt.x<=n&&nxt.y>=1&&nxt.y<=m){
				if(d[nxt.x][nxt.y]==-1){
					d[nxt.x][nxt.y]=d[now.x][now.y]+1;
					q.push(nxt);
				}
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			cout<<d[i][j]<<" ";
		}
		cout<<'\n';
	}
}

int main(){
	cin>>n>>m;
	memset(d,-1,sizeof(d));
	for(int i=1;i<=n;i++){
		cin>>s[i]+1;
	}
	bfs();
	return 0;
}