FZU1408（BFS+优化） 之 位图

 Problem 1408 位图

Accept: 502    Submit: 2281
Time Limit: 1000 mSec    Memory Limit : 32768 KB

Problem Description

现在我们给出一个n*m的黑白位图，n为行数，m为列数，且该位图中至少含有一个白色的像素。我们用(i,j)来表示第i行第j列的像素，并且定义两像素点p1=(i1,j1)和p2=(i2,j2)之间的距离为：d(p1,p2)=|i1-i2|+|j1-j2|。

对于任意给定的位图，计算出位图中每个像素到与它最近的白色像素之间的距离。

Input

本题有多组输入数据，每组数据第一行包含两个整数n,m，用空格分开，1<=n,m<=182。之后的n行每行包含一个长度为m的由0和1组成的字符串，即第i行第j个字符如果为"1"，那么表示像素(i,j)为白的，否则为黑的。

Output

对于每组数据，输出包含n行，第i行的第j个数字为f(i,j)，表示像素(i,j)到最近的白色像素的距离。每行内的数字用空格分开。

Sample Input

3 4

0001

0011

0110

Sample Output

3 2 1 0

2 1 0 0

1 0 0 1 

分析：广度优先搜索，刚开始写的代码老师内存超限，看了讨论知道，不能从每个0去找1，而应该从1去找0，这样比较节约内存和时间

memory limit exceed 代码：

#include <iostream>#include <cstdio>#include <cstring>#include <queue>using namespace std;const int maxn=200;int flag[200][200];char a[200][200];int n,m;int direction[4][2]={{0,-1},{0,1},{-1,0},{1,0}};int ans[40000],num;struct point{    char value;    int x,y;    int step;};void bfs(char value,int row,int col,int step){    int x,y;    point v,w,top;    v.value=value;    v.x=row;    v.y=col;    v.step=step;    queue<point> q;    q.push(v);    flag[row][col]=step+1;    while(!q.empty())    {        top=q.front(); q.pop();        if(top.value=='1')        {            ans[num]=top.step;            num++;            break;        }        for(int i=0;i<4;i++)        {            x=top.x+direction[i][0];            y=top.y+direction[i][1];            if(x>=0 && x<n && y>=0 && y<m && !flag[x][y])            {                w.value=a[x][y];                w.x=x;                w.y=y;                w.step=top.step+1;                q.push(w);            }        }    }}int main(){    while(cin>>n>>m)    {        num=0;        memset(ans,0,sizeof(ans));        for(int i=0; i<n; i++)            scanf("%s",a[i]);        for(int i=0; i<n; i++)        {            for(int j=0; j<m; j++)            {                if(a[i][j]=='1')                {                    ans[num]=0;                    num++;                }                else                {                    memset(flag,0,sizeof(flag));                    bfs(a[i][j],i,j,0);                }            }        }        for(int i=0; i<num; i++)        {            cout<<ans[i];            if((i+1)%m==0) cout<<endl;            else cout<<" ";        }    }    return 0;}

AC代码如下：

#include <iostream>#include <cstdio>#include <cstring>#include <queue>using namespace std;const int maxn=200;int flag[200][200];char a[200][200];int n,m;int direction[4][2]={{0,-1},{0,1},{-1,0},{1,0}};int ans[200][200];struct point{    char value;    int x,y;    int step;};queue<point> q;void bfs(){    while(!q.empty())    {        point top=q.front(); q.pop();        for(int i=0;i<4;i++)        {            int x=top.x+direction[i][0];            int y=top.y+direction[i][1];            if(x>=0 && x<n && y>=0 && y<m && !flag[x][y])            {                flag[x][y]=1;                ans[x][y]=top.step+1;                point w;                w.value=a[x][y];                w.x=x;                w.y=y;                w.step=top.step+1;                q.push(w);            }        }    }}int main(){    while(cin>>n>>m)    {        point cur;        memset(ans,0,sizeof(ans));        memset(flag,0,sizeof(flag));        for(int i=0; i<n; i++)            scanf("%s",a[i]);        for(int i=0; i<n; i++)        {            for(int j=0; j<m; j++)            {                if(a[i][j]=='1')                {                    flag[i][j]=1;                    ans[i][j]=0;                    cur.value='1';                    cur.x=i;                    cur.y=j;                    cur.step=0;                    q.push(cur);                }            }        }        bfs();        for(int i=0; i<n; i++)        {            for(int j=0; j<m; j++)            {                cout<<ans[i][j];                if(j!=m-1) cout<<" ";            }            cout<<endl;        }    }    return 0;}