HihoCoder - 1478 水陆距离

                                               水陆距离

                                                               Time limit                              10000 ms

                                                             Memory limit                           262144 kB

 

给定一个N x M的01矩阵，其中1表示陆地，0表示水域。对于每一个位置，求出它距离最近的水域的距离是多少。  

矩阵中每个位置与它上下左右相邻的格子距离为1。

Input

第一行包含两个整数，N和M。

以下N行每行M个0或者1，代表地图。

数据保证至少有1块水域。

对于30%的数据，1 <= N, M <= 100  

对于100%的数据，1 <= N, M <= 800

Output

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

Sample Input

4 4  
0110  
1111  
1111  
0110

Sample Output

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

 

 

本题很容易看出来是广搜，想法就是直接BFS解决这道题。

#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
#include <stack>
#include<vector>
using namespace std;
short int mapp[1811][8111];
short int a[1811][8111];
char b[1811][8111];
short int mov[4][2]={0,1,0,-1,1,0,-1,0};
short int indexxx[811][811];
int n,m;
const int inf=99999999;
struct xiao
{
    short int x,y;
}mmm[8101];

void bfs(int x,int y,int temp)
{
    if(temp>n*m) return ;
    if(x<0||x>=n||y<0||y>=m) return ;
    if(a[x][y]>temp)
        a[x][y]=temp;
    for(int i=0;i<4;++i)
    {
        int fx=x+mov[i][0];
        int fy=y+mov[i][1];
        if(fx<0||fx>=n||fy<0||fy>=m||indexxx[fx][fx]) continue;
        indexxx[fx][fy]=1;
        bfs(fx,fy,temp+1);
        indexxx[fx][fy]=0;
    }
}
int main()
{
    int cnt=0;
    scanf("%d%d",&n,&m);
        for(int j=0;j<n;++j)
           {
               scanf("%s",&b[j]);
               for(int i=0;i<m;++i)
               {
                a[j][i]=b[j][i]-'0';
                if(b[j][i]=='0')
               {
                   mmm[cnt].x=j;
                   mmm[cnt++].y=i;
               }
               else
               {
                   a[j][i]=inf;
               }
               }
           }
    for(int i=0;i<cnt;++i)
    {
        bfs(mmm[i].x,mmm[i].y,0);
    }
    for(int i=0;i<n;++i)
    {
        for(int j=0;j<m;++j)
        {
            printf("%d ",a[i][j]);
        }
        printf("\n");
    }
    return 0;
}

但是，这样做时间超限了！

 

这道题的数据范围是800*800，如果说这些数据全部都是0的话，我们遍历一遍所需要的操作是800*800*800，而给出的时间限制是1s，所以这道题要优化。

简单优化

1、将单个点查找比较大小改变为多个点同时查找

2、输入操作不用将字符数组改变成整形数组

 

小补充：可以通过这个操作实现从1开始输入。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<queue>
#include<stack>
using namespace std;
int n,m;
char mappp[805][805];
int mov[4][2]={0,1,0,-1,1,0,-1,0};
int indexx[805][805];
struct xiao
{
    int x,y,z;
}p,a;

int main()
{
    queue<xiao>q;
    cin>>n>>m;
    memset(indexx,-1,sizeof(indexx));
    for(int i=0;i<n;++i)
    {
            scanf("%s",&mappp[i]);
            for(int j=0;j<m;++j)
            {
                if(mappp[i][j]=='0')
                {
                    p.x=i;
                    p.y=j;
                    p.z=0;
                    indexx[p.x][p.y]=0;
                    q.push(p);
                }
            }
    }
    while(!q.empty())//如果说不为空的话
    {
        a=q.front();
        q.pop();//多个点同时遍历
        for(int i=0;i<4;++i)
        {
            p.x=a.x+mov[i][0];
            p.y=a.y+mov[i][1];
            p.z=a.z+1;
            if(p.x<0||p.x>=n||p.y<0||p.y>=m)
                continue;
            if(indexx[p.x][p.y]==-1)//如果这一点没有经过
             indexx[p.x][p.y]=p.z,q.push(p);
        }
    }
    for(int i=0;i<n;++i)
    {
        for(int j=0;j<m;++j)
        {
            printf("%d ",indexx[i][j]);
        }
        printf("\n");
    }
    return 0;
}