BZOJ2252 [2010Beijing wc]矩阵距离

2252: [2010Beijing wc]矩阵距离

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 624  Solved: 304
[Submit][Status][Discuss]

Description

 

假设我们有矩阵,其元素值非零即1

a11…… a1m

…………….

an1…….anm

 

定义aij与akl之间的距离为D(aij,akl)=abs(i-k)+abs(j-L) 

Input

输入文件的第一行为两个整数，分别代表n和m。
接下来的n行，第i行的第 j个字符代表aij

Output

输出包含N行，每行M个用空格分开的数字，其中第i行第J个数字代表
Min(D(aij,axy) 1<=x<=N 1<=y<m,且axy=1

Sample Input

3 4
0001
0011
0110

Sample Output

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

HINT

对于100%的数据，满足 0 <  m n <=1000

Source

 

【题解】

题面凑活着看吧。

还是需要一点小小的转换。

min(dist(A[i][j],A[x][y])其实就是(i,j)到最近的'1'的点的曼哈顿距离

BFS即可

初始状态是所有为1的点

 

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>

const int MAXN = 1000 + 10;
const int MAXM = 1000 + 10;
const int dx[4] = {0,0,1,-1};
const int dy[4] = {1,-1,0,0};

inline void read(int &x)
{
    x = 0;char ch = getchar(), c = ch;
    while(ch < '0' || ch > '9')c = ch, ch = getchar();
    while(ch <= '9' &&ch >= '0')x = x * 10 + ch - '0', ch = getchar();
    if(c == '-')x = -x;
}

int n,m,g[MAXN][MAXM],ans[MAXN][MAXM],b[MAXN][MAXM];
char s[MAXM];

struct Node
{
    int x, y;
    Node(int xx, int yy){x = xx;y = yy;}
    Node(){}
};

std::queue<Node> q;

void bfs()
{
    register Node tmp;
    register int xx, yy;
    while(q.size())
    {
        tmp = q.front();q.pop();
        for(register int i = 0;i < 4;++ i)
        {
            xx = tmp.x + dx[i], yy = tmp.y + dy[i];
            if(xx < 1 || yy < 1 || xx > n || yy > m || b[xx][yy])continue;
            ans[xx][yy] = ans[tmp.x][tmp.y] + 1;
            b[xx][yy] = 1;
            q.push(Node(xx, yy));
        }
    }
}

int main()
{
    read(n), read(m);
    for(register int i = 1;i <= n;++ i)
        {
            scanf("%s", s + 1);
            for(register int j = 1;j <= m;++ j)
            {
                g[i][j] = s[j] - '0';
                if(g[i][j])
                {
                    b[i][j] = 1;
                    q.push(Node(i,j));
                }
            }
        }
    bfs();
    for(register int i = 1;i <= n;++ i)
    {
        for(register int j = 1;j <= m;++ j)
            printf("%d ", ans[i][j]);
        putchar('\n');
    }
    return 0;
}