BFS(多源) - 矩阵距离 - 小马智行面试题

BFS - 矩阵距离 - 小马智行面试题

$给定一个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]=mi{n}_{1\le x\le N,1\le y\le M,A[x][y]=1}dist(A[i][j],A[x][y])$

输入格式
第一行两个整数n,m。

接下来一个N行M列的01矩阵，数字之间没有空格。

输出格式
一个N行M列的矩阵B，相邻两个整数之间用一个空格隔开。

数据范围

$1\le N,M\le 1000$

输入样例：

3 4
0001
0011
0110

输出样例：

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

---

分析：

$题意要求矩阵中每一{个}^{\prime }{0}^{\prime }到最近{的}^{\prime }{1}^{\prime }的距离。$

$若对每一{个}^{\prime }{0}^{\prime }都跑一遍最短距离，或者对每一{个}^{\prime }{1}^{\prime }跑一遍全图更新最短距离均会导致TLE。$

$正解是先将所有的1入队作为起点，然后整体跑一遍BFS。$

$可以将矩阵中所有{的}^{\prime }{1}^{\prime }视作第一层节点，接着向外拓展。对每一{个}^{\prime }{0}^{\prime }而言，第一次出队的距离就是最短距离。$

$因为队列中最多仅有2层节点，距离是单调递增的，第一次拓展到的就是最短距离。$

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>

#define P pair<int,int>
#define x first
#define y second

using namespace std;

const int N=1010;

int n,m;
char g[N][N];
P q[N*N];
int d[N][N];
int dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}};

int bfs()
{
    memset(d,-1,sizeof d);
    int hh=0,tt=-1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(g[i][j]=='1')
            {
                d[i][j]=0;
                q[++tt]=make_pair(i,j);
            }

    while(hh<=tt)
    {
        P t=q[hh++];

        for(int i=0;i<4;i++)
        {
            int x=t.x+dir[i][0],y=t.y+dir[i][1];
            if(x<1||x>n||y<1||y>m||d[x][y]!=-1) continue;
            
            d[x][y]=d[t.x][t.y]+1;
            q[++tt]=make_pair(x,y);
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%s",g[i]+1);

    bfs();

    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            printf("%d ",d[i][j]);
        puts("");
    }
    
    return 0;
}