[Programmers] 게임 맵 최단거리

게임 맵 최단거리

문제 설명

게임 맵의 상태 maps가 매개변수로 주어질 때, 캐릭터가 상대 팀 진영에 도착하기 위해서 지나가야 하는 칸의 개수의 최솟값을 return 하도록 solution 함수를 완성해주세요. 단, 상대 팀 진영에 도착할 수 없을 때는 -1을 return 해주세요.

제한사항

maps는 n x m 크기의 게임 맵의 상태가 들어있는 2차원 배열로, n과 m은 각각 1 이상 100 이하의 자연수입니다.
n과 m은 서로 같을 수도, 다를 수도 있지만, n과 m이 모두 1인 경우는 입력으로 주어지지 않습니다.
maps는 0과 1로만 이루어져 있으며, 0은 벽이 있는 자리, 1은 벽이 없는 자리를 나타냅니다.
처음에 캐릭터는 게임 맵의 좌측 상단인 (1, 1) 위치에 있으며, 상대방 진영은 게임 맵의 우측 하단인 (n, m) 위치에 있습니다.

풀이

전형적인 BFS 문제

코드

#include<bits/stdc++.h>
using namespace std;

int dx[4] = {1,0,-1,0};
int dy[4] = {0,1,0,-1};
int dist[101][101];

int solution(vector<vector<int> > maps){
    int n = maps.size();
    int m = maps[0].size();
    for(int i=0; i<n; i++)
        for(int j=0; j<m; j++)
            dist[i][j]=-1;
    
    dist[0][0]=1; // start
    queue<pair<int, int>> q;
    q.push({0,0});
    while(!q.empty()){
        auto cur = q.front(); q.pop();
        int x=cur.first, y=cur.second;
        for(int k=0; k<4; k++){
            int nx=x+dx[k], ny=y+dy[k];
            if(nx<0 || nx>=n || ny<0 || ny>=m) continue; // OOB
            if(dist[nx][ny]==-1 && maps[nx][ny]==1){
                dist[nx][ny] = dist[x][y]+1;
                q.push({nx,ny});
            }
        }
    }
    
    return dist[n-1][m-1];
}

정리