Home > Algorithm > BOJ > [BOJ] 14940번 쉬운 최단거리 / C++

[BOJ] 14940번 쉬운 최단거리 / C++
C++ 백준 BOJ 알고리즘 Algorithm

🔗 쉬운 최단거리

https://www.acmicpc.net/problem/14940


📌 문제 요약

지도가 주어지면 모든 지점에 대해서 목표지점까지의 거리를 구하여라.
문제를 쉽게 만들기 위해 오직 가로와 세로로만 움직일 수 있다고 하자.

입력: 지도의 크기 n과 m이 주어진다. n은 세로의 크기, m은 가로의 크기다.(2 ≤ n ≤ 1000, 2 ≤ m ≤ 1000)
다음 n개의 줄에 m개의 숫자가 주어진다. 0은 갈 수 없는 땅이고 1은 갈 수 있는 땅, 2는 목표지점이다. 입력에서 2는 단 한개이다. **
**출력: 각 지점에서 목표지점까지의 거리를 출력한다. 원래 갈 수 없는 땅인 위치는 0을 출력하고, 원래 갈 수 있는 땅인 부분 중에서 도달할 수 없는 위치는 -1을 출력한다.


💡 접근 방법

queue를 사용한 bfs알고리즘 방식으로 문제를 풀어냈다.
queue에 목표지점을 저장한 뒤 주변 네방향의 좌표가 1인 경우 push 해준다.
네방향의 거리는 현재 거리에서 1을 더한것과 같기 때문에 push해줌과 동시에 더해준다.


⚠️ 처음에 했던 실수


💻 코드

#include <iostream>
#include <cmath>
#include <utility>
#include <queue>
using namespace std;

int n, m;
int map[1001][1001];
int dis[1001][1001];

int offset[4][2] =
{
    {1,0},
    {0,1},
    {-1,0},
    {0,-1},
};

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    pair<int, int> fin;
    cin >> n >> m;
    for(int i = 0; i < n; ++i)
    {
        for(int j = 0; j < m; ++j)
        {
            cin >> map[i][j];
            if(map[i][j] == 2)
            {
                fin.first = i;
                fin.second = j;
            }
        }
    }
    queue<pair<int, int>> q;
    q.push(fin);
    int cnt = 0;
    dis[fin.first][fin.second] = cnt;
    while(q.size() != 0)
    {
        pair<int, int> f = q.front();
        q.pop();
        for(int i = 0; i < 4; ++i)
        {
            int nexR = f.first + offset[i][0];
            int nexC = f.second + offset[i][1];

            if(nexR < 0 || nexC < 0 || nexR >= n || nexC >= m)
                continue;
        
            if(fin.first == nexR && fin.second == nexC)
                continue;
        
            if(map[nexR][nexC] == 0)
            {
                dis[nexR][nexC] = 0;
                continue;
            }

            if(dis[nexR][nexC] > 0)
                continue;
            
            dis[nexR][nexC] = dis[f.first][f.second] + 1;
            q.push(make_pair(nexR, nexC));
        }
    }

    for(int i = 0; i < n; ++i)
    {
        for(int j = 0; j < m; ++j)
        {
            if(dis[i][j] == 0 && map[i][j] == 1)
            {
                dis[i][j] = -1;
            }

            cout << dis[i][j] << " ";
        }
        
        cout << '\n';
    }
    return 0;
}