PIPIOJ 1032: 水陆距离

1032: 水陆距离

题目描述

给定一个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，只是要注意的陆地到水的距离就是水到陆地的距离，对于所有水域同时开始 BFS，可以降低时间复杂度。若对于所有陆地都进行搜索，可能会超时。

#include<bits/stdc++.h>
using namespace std;
char a[810][810];
int vis[810][810];
int dir[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
int ans[810][810];
typedef struct Node
{
	int x, y, t;
}node;
int main()
{
	int n, m, len;
	while(scanf("%d%d", &n, &m) != EOF) 
	{
		memset(vis, 0, sizeof(vis));
		memset(ans, 0, sizeof(ans));
		queue<node> q;
		for(int i = 0; i < n; i++)
		{
			scanf("%s", a[i]);
			for(int j = 0; j < m; j++)
				if(a[i][j] == '0')
				{
					q.push({i, j, 0});
					vis[i][j] = 1;
				}		
		}
		node p;
		while(!q.empty())
		{
			len = q.size();
			while(len--)
			{
				p = q.front();
				q.pop();
				for(int i = 0; i < 4; i++)
				{
					int x = p.x + dir[i][0], y = p.y + dir[i][1];
					if(x >= 0 && x < n && y >= 0 && y < m && !vis[x][y])
					{
						q.push({x, y, p.t + 1});
						ans[x][y] = p.t + 1;
						vis[x][y] = 1;
					}
				}
			}
		}
		for(int i = 0; i < n; i++)	
		{
			for(int j = 0; j < m; j++)
				printf("%d ", ans[i][j]);
			printf("\n");
		}
	}
	return 0;
}