[백준] 14940번 : 쉬운 최단거리 (C++)

14940번 : 쉬운 최단거리

---

 

문제)

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

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

 

 

 

입력 :

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

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

 

 

 

출력 :

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

 

 

 

 

 

 

풀이)

 

#include <iostream>
#include <vector>
#include <queue>
 
using namespace std;
 
int n, m;
vector<vector<int>> vec;
vector<vector<int>> visited;
 
int dx[] = { -1, 0, 1, 0 };
int dy[] = { 0, -1, 0, 1 };
 
void bfs(int a, int b)
{
    queue<pair<int, int>> q;
    visited[a][b] = 1;
    q.push(make_pair(a, b));
 
    while (!q.empty())
    {
        int x = q.front().first;
        int y = q.front().second;
        q.pop();
 
        for (int i = 0; i < 4; i++)
        {
            int nx = x + dx[i];
            int ny = y + dy[i];
 
            if (0 > nx || nx >= n) continue;
            if (0 > ny || ny >= m) continue;
            if (visited[nx][ny]) continue;
            if (vec[nx][ny] == 0)
            {
                visited[nx][ny] = -1;
                continue;
            }
            if (vec[nx][ny] == 1)
            {
                visited[nx][ny] = visited[x][y] + 1;
                q.push(make_pair(nx, ny));
            }
        }
    }
 
}
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
 
    cin >> n >> m;
 
    vec.resize(n, vector<int>(m));
    visited.resize(n, vector<int>(m, 0));
 
    int t;
 
    pair<int, int> start;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            cin >> t;
            if (t == 2)
            {
                start.first = i;
                start.second = j;
            }
            vec[i][j] = t;
        }
    }
 
    bfs(start.first, start.second);
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            // 원래 갈 수 없는 땅
            if (vec[i][j] == 0)
            {
                cout << "0 ";
            }
            else
            {    
                if (vec[i][j] == 1 && visited[i][j] == 0)
                {    // 갈 수 있는데 못간 땅
                    cout << "-1 ";
                }
                else
                {    // 나머지 땅들
                    cout << visited[i][j] - 1 << " ";
                }
            }
        }
        cout << "\n";
    }
 
    return 0;
}
 
 
//15 15
//2 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 0 0 0 0 0
//1 1 1 1 1 1 1 1 1 1 0 1 1 1 1
//1 1 1 1 1 1 1 1 1 1 0 1 0 0 0
//1 1 1 1 1 1 1 1 1 1 0 1 1 1 1

---

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

14940번: 쉬운 최단거리