多源BFS之矩阵距离

多源BFS

173. 矩阵距离

给定一个 N行 M列的 01矩阵 A,A[i][j]与 A[k][l]之间的曼哈顿距离定义为dist(i,j,k,l)=|i−k|+|j−l|

输出一个 N行 M列的整数矩阵 B,其中: B[i][j]=min1≤x≤N,1≤y≤M,A[x][y]=1dist(i,j,x,y)

输入格式

第一行两个整数 N,M。

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

输出格式

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

数据范围

1≤N,M≤1000

输入样例:

3 4

0001

0011

0110

输出样例:

3 2 1 0

2 1 0 0

1 0 0 1

思路

全部位置为1的地方设置成dist=0,全部入队依次扩展。

cpp 复制代码
#include<iostream>
#include<cstring>
#include<algorithm>

#define x first
#define y second

using namespace std;

typedef pair<int,int> pii;
const int N = 1010, M = N*N;

int n,m;
char g[N][N];
pii q[M];
int dist[N][N];

void bfs(){
    memset(dist,-1,sizeof dist);
    int hh=0,tt=-1;//插个锚点,后面的操作中先加tt后赋值,少了tt=0,这个操作
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(g[i][j]=='1'){
                dist[i][j]=0;
                q[++tt]={i,j};
            }
        }
    }
    int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
    while(hh<=tt){
        pii t=q[hh++];
        for(int i=0;i<4;i++){
            int a=t.x+dx[i],b=t.y+dy[i];
            if(a<0||a>=n||b<0||b>=m) continue;
            if(dist[a][b]!=-1) continue;
            
            dist[a][b]=dist[t.x][t.y]+1;
            q[++tt]={a,b};
        }
    }
    
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) scanf("%s",g[i]);
    bfs();
    for(int i=0;i<n;i++){
      for(int j=0;j<m;j++)
        printf("%d ",dist[i][j]);
    printf("\n");
    }
    return 0;
}
相关推荐
潇湘夜雨6976 分钟前
2023ICPC合肥题解
算法
Small踢倒coffee_氕氘氚10 分钟前
iPhone闹钟无法识别调休致用户迟到,苹果客服称会记录反馈
笔记·算法·灌灌灌灌
制冷男孩14 分钟前
机器学习算法-支持向量机SVM
人工智能·算法·机器学习·支持向量机
晨曦54321022 分钟前
Numpy数组与矩阵——python学习
python·矩阵·numpy
代码程序猿RIP29 分钟前
【C语言干货】回调函数
c语言·开发语言·数据结构·c++·算法
mljy.1 小时前
递归、搜索和回溯算法《递归》
算法
知识漫步1 小时前
代码随想录算法训练营第60期第二十一天打卡
数据结构·算法
刃神太酷啦1 小时前
排序--数据结构初阶(4)(C/C++)
c语言·数据结构·c++·算法·leetcode·深度优先·广度优先
凯子坚持 c1 小时前
深度解析算法之分治(归并)
算法·leetcode·职场和发展
是店小二呀1 小时前
【优选算法-二分查找】二分查找算法解析:如何通过二段性优化搜索效率
c++·算法