Acwing844.走迷宫

使用STL的queue实现

#include
#include
#include
#include

using namespace std;

typedef pair<int,int> PII;
const int N = 110;

int n,m;
queue<PII> q;
int grid[N][N];
// 保存距离
int d[N][N];

int bfs()
{
    // 初始化操作
    q.push({0,0});
    memset(d,-1,sizeof d);
    d[0][0] = 0;
    
    // 使用向量的方式枚举
    int dx[4] = {-1,0,1,0}, dy[4] = {0,1,0,-1};
    
    while(!q.empty())
    {
        auto e = q.front();
        q.pop();
        for(int i = 0;i < 4;i++)
        {
            int x = e.first + dx[i], y = e.second + dy[i];
            if(x < n && x >= 0 && y < m && y >= 0 && d[x][y] == -1 && grid[x][y] == 0)
            {
                d[x][y] = d[e.first][e.second] + 1;
                q.push({x,y});
            }
        }
    }
    return d[n-1][m-1];
}

int main()
{
    cin >> n >> m;
    for(int i = 0;i < n;i++)
        for(int j = 0;j < m;j++)
            cin >> grid[i][j];
    cout << bfs() << endl;
}

你可能感兴趣的:(C++,acwing,bfs)