Acwing--173. 矩阵距离（多源BFS）

 173. 矩阵距离 - AcWing题库

 

宽搜由于每个点都搜一遍，所以时间复杂度是O（n）

这个题属于多源最短路问题。

我们可以假设个虚拟原点,转换为单源最短路径

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int>PII;
char g[1010][1010];
int d[1010][1010];
int dx[4]={0,0,-1,1};
int dy[4]={1,-1,0,0};
int n,m;

void bfs()
{
    queue<PII>q;
   // q.push({})
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(g[i][j]=='1')
            {
                d[i][j]=0;
                q.push({i,j});
            }
            
        }
    }
    
    while(q.size())
    {
        PII t=q.front();
        q.pop();
        for(int i=0;i<4;i++)
        {
            int nx=dx[i]+t.first;
            int ny=dy[i]+t.second;
            
            if(nx<0||ny<0||nx>=n||ny>=m)
            continue;
            if(d[nx][ny]!=-1)
            continue;
            
            d[nx][ny]=d[t.first][t.second]+1;
            q.push({nx,ny});
        }
    }
    
}

int main()
{
    memset(d,-1,sizeof(d));
    
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            cin>>g[i][j];
        }
    }
    
    bfs();
    
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            cout<<d[i][j]<<" ";
        }
        cout<<endl;
    }
    
    
    
    return 0;
}