BFS - 矩阵距离 - 小马智行面试题
给 定 一 个 N 行 M 列 的 01 矩 阵 A , A [ i ] [ j ] 与 A [ k ] [ l ] 之 间 的 曼 哈 顿 距 离 定 义 为 : 给定一个N行M列的01矩阵A,A[i][j] 与 A[k][l] 之间的曼哈顿距离定义为: 给定一个N行M列的01矩阵A,A[i][j]与A[k][l]之间的曼哈顿距离定义为:
d i s t ( A [ i ] [ j ] , A [ k ] [ l ] ) = ∣ i − k ∣ + ∣ j − l ∣ dist(A[i][j],A[k][l])=|i−k|+|j−l| dist(A[i][j],A[k][l])=∣i−k∣+∣j−l∣
输 出 一 个 N 行 M 列 的 整 数 矩 阵 B , 其 中 : 输出一个N行M列的整数矩阵B,其中: 输出一个N行M列的整数矩阵B,其中:
B [ i ] [ j ] = m i n 1 ≤ x ≤ N , 1 ≤ y ≤ M , A [ x ] [ y ] = 1 d i s t ( A [ i ] [ j ] , A [ x ] [ y ] ) B[i][j]=min_{1≤x≤N,1≤y≤M,A[x][y]=1}dist(A[i][j],A[x][y]) B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(A[i][j],A[x][y])
输入格式
第一行两个整数n,m。
接下来一个N行M列的01矩阵,数字之间没有空格。
输出格式
一个N行M列的矩阵B,相邻两个整数之间用一个空格隔开。
数据范围
1 ≤ N , M ≤ 1000 1≤N,M≤1000 1≤N,M≤1000
输入样例:
3 4
0001
0011
0110
输出样例:
3 2 1 0
2 1 0 0
1 0 0 1
分析:
题 意 要 求 矩 阵 中 每 一 个 ′ 0 ′ 到 最 近 的 ′ 1 ′ 的 距 离 。 题意要求矩阵中每一个'0'到最近的'1'的距离。 题意要求矩阵中每一个′0′到最近的′1′的距离。
若 对 每 一 个 ′ 0 ′ 都 跑 一 遍 最 短 距 离 , 或 者 对 每 一 个 ′ 1 ′ 跑 一 遍 全 图 更 新 最 短 距 离 均 会 导 致 T L E 。 若对每一个'0'都跑一遍最短距离,或者对每一个'1'跑一遍全图更新最短距离均会导致TLE。 若对每一个′0′都跑一遍最短距离,或者对每一个′1′跑一遍全图更新最短距离均会导致TLE。
正 解 是 先 将 所 有 的 1 入 队 作 为 起 点 , 然 后 整 体 跑 一 遍 B F S 。 正解是先将所有的1入队作为起点,然后整体跑一遍BFS。 正解是先将所有的1入队作为起点,然后整体跑一遍BFS。
可 以 将 矩 阵 中 所 有 的 ′ 1 ′ 视 作 第 一 层 节 点 , 接 着 向 外 拓 展 。 对 每 一 个 ′ 0 ′ 而 言 , 第 一 次 出 队 的 距 离 就 是 最 短 距 离 。 可以将矩阵中所有的'1'视作第一层节点,接着向外拓展。对每一个'0'而言,第一次出队的距离就是最短距离。 可以将矩阵中所有的′1′视作第一层节点,接着向外拓展。对每一个′0′而言,第一次出队的距离就是最短距离。
因 为 队 列 中 最 多 仅 有 2 层 节 点 , 距 离 是 单 调 递 增 的 , 第 一 次 拓 展 到 的 就 是 最 短 距 离 。 因为队列中最多仅有2层节点,距离是单调递增的,第一次拓展到的就是最短距离。 因为队列中最多仅有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;
}