矩阵距离——多源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]=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

输入样例:

3 4

0001

0011

0110

输出样例:

3 2 1 0

2 1 0 0

1 0 0 1

解析:

将 "1" 点放入队列,再遍历即可。不过要注意输入的问题,要用字符数组输入。

复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long 
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
typedef pair<int,int> PII;
const int N=2e6+10;
char g[1010][1010];
int d[1010][1010];
bool vis[1010][1010];
int dx[4]={-1,1,0,0};
int dy[4]={0,0,-1,1};
int n,m;
queue <PII> q;
void bfs()
{
    for (int i=0;i<n;i++)
    for (int j=0;j<m;j++)
    {
        if (g[i][j]=='1')
        {
            q.push({i,j});
            vis[i][j]=1;
        }
    }
    while (q.size())
    {
        int x=q.front().first;
        int y=q.front().second;
        q.pop();
        for (int i=0;i<4;i++)
        {
            int a=x+dx[i],b=y+dy[i];
            if (a>=0&&a<n&&b>=0&&b<m&&!vis[a][b])
            {
                d[a][b]=d[x][y]+1;
                q.push({a,b});
                vis[a][b]=1;
            }
        }
    }
}
signed main()
{
    ios;
    cin>>n>>m;
    for (int i=0;i<n;i++) cin>>g[i];
    bfs();
    for (int i=0;i<n;i++)
    {
        for (int j=0;j<m;j++) cout<<d[i][j]<<" ";
        cout<<endl;
    }
    return 0;
}
相关推荐
C嘎嘎嵌入式开发5 分钟前
(10)100天python从入门到拿捏《Python中的数据结构与自定义数据结构》
数据结构·python·算法
熬了夜的程序员6 分钟前
【LeetCode】69. x 的平方根
开发语言·算法·leetcode·职场和发展·动态规划
Niuguangshuo14 分钟前
音频特征提取算法介绍
算法·音视频
Vect__20 分钟前
从底层到上层的“外挂”:deque、stack、queue、priority_queue 全面拆解
数据结构·c++
fengfuyao98529 分钟前
基于MATLAB的匈牙利算法实现任务分配
算法·数学建模·matlab
CoovallyAIHub35 分钟前
超详细链式插补 (MICE) 多元插补:机器学习模型的高级缺失数据处理
算法·机器学习·计算机视觉
明天会有多晴朗1 小时前
C语言入门教程(第6讲):函数——让程序学会“分工合作”的魔法
c语言·开发语言·算法
玖釉-1 小时前
三维模型数据结构与存储方式解析
数据结构·算法·图形渲染
草莓熊Lotso1 小时前
《算法闯关指南:优选算法--二分查找》--17.二分查找(附二分查找算法简介),18. 在排序数组中查找元素的第一个和最后一个位置
开发语言·c++·算法
努力努力再努力wz1 小时前
【C++进阶系列】:万字详解特殊类以及设计模式
java·linux·运维·开发语言·数据结构·c++·设计模式