[BOJ / C++] 14940번 : 쉬운 최단거리

문제

지도가 주어지면 모든 지점에 대해서 목표지점까지의 거리를 구하여라.

문제를 쉽게 만들기 위해 오직 가로와 세로로만 움직일 수 있다고 하자.

입력

지도의 크기 n과 m이 주어진다. n은 세로의 크기, m은 가로의 크기다.(2 ≤ n ≤ 1000, 2 ≤ m ≤ 1000)

다음 n개의 줄에 m개의 숫자가 주어진다. 0은 갈 수 없는 땅이고 1은 갈 수 있는 땅, 2는 목표지점이다. 입력에서 2는 단 한개이다.

출력

각 지점에서 목표지점까지의 거리를 출력한다. 원래 갈 수 없는 땅인 위치는 0을 출력하고, 원래 갈 수 있는 땅인 부분 중에서 도달할 수 없는 위치는 -1을 출력한다.

Solve.ac 레벨

실버 I

풀이

#include <bits/stdc++.h>
#define MAX 1001

using namespace std;

int n, m;
int dx[4] = {0, 0, -1, 1};
int dy[4] = {-1, 1, 0, 0};
queue<pair<int, int> > q;
int dist[MAX][MAX];
bool visited[MAX][MAX];
int node[MAX][MAX];


void BFS(pair<int, int> p) {
    dist[p.first][p.second] = 0;
    visited[p.first][p.second] = true;
    q.push(p);

    while(!q.empty()) {
        pair<int, int> now = q.front();
        q.pop();

        for(int i = 0; i < 4; i++) {
            int nextX = now.first + dx[i];
            int nextY = now.second + dy[i];

            if(nextX < 0 || nextX >= n || nextY < 0 || nextY >= m) continue;
            if(node[nextX][nextY] == 0) continue;
            if(visited[nextX][nextY]) continue;

            q.push(make_pair(nextX, nextY));
            dist[nextX][nextY] = dist[now.first][now.second] + 1;
            visited[nextX][nextY] = true;
        }
    }


}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    pair<int, int> des;

    cin >> n >> m;

    for(int i = 0; i < n; i++) {
        for(int j = 0; j < m; j++) {
            cin >> node[i][j];

            if(node[i][j] == 2) {
                des.first = i;
                des.second = j;
            }
        }
    }

    BFS(des);

    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++) {
            if(dist[i][j] == 0 && node[i][j] == 1){
                cout << -1 << " ";
            } else {
                cout << dist[i][j] << " ";
            }
        }
        cout << "\n";
    }
    return 0;
}

 

14940번: 쉬운 최단거리