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

문제

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

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

---

입력

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

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

출력

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

---

예제 입력 1 복사

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 1
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

예제 출력 1 복사

0 1 2 3 4 5 6 7 8 9 10 11 12 13 14
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
11 12 13 14 15 16 17 18 19 20 0 0 0 0 25
12 13 14 15 16 17 18 19 20 21 0 29 28 27 26
13 14 15 16 17 18 19 20 21 22 0 30 0 0 0
14 15 16 17 18 19 20 21 22 23 0 31 32 33 34

---

풀이과정

문제 자체는 크게 어렵지않게 접근할 수 있는 문제인 거 같다.

입력값에서 2라고 적힌 부분부터 출발을 해서 1 이라고 적힌 모든 지점을 다 갔을 때 2와 1의 거리차이가 얼마나 걸리는가 라는 문제라서 문제에 대한 이해자체가 어렵지는 않았는데 어떻게 접근을 해야할지가 막막 했던거 같다.

 

code

#include <iostream>
#include <queue>

int		map_scan(int map[1001][1001], int new_map[1001][1001], int n, int m, int start_y, int start_x)
{
	int	map_check_y[4] = {0, 0, -1, 1};
	int	map_check_x[4] = {1, -1, 0, 0};
	std::queue<std::pair<int, int> > q;

	q.push(std::make_pair(start_y, start_x));
	new_map[start_y][start_x] = 0;
	while(!q.empty())
	{
		std::pair<int, int> temp = q.front();
		q.pop();
		for(int i = 0; i < 4; i++) {
			int temp_y = temp.first + map_check_y[i];
			int temp_x = temp.second + map_check_x[i];
			if (0 > temp_y || 0 > temp_x || temp_y >= n || temp_x >= m) continue;
			if (new_map[temp_y][temp_x] != -1 || map[temp_y][temp_x] != 1) continue;
			new_map[temp_y][temp_x] = new_map[temp.first][temp.second] + 1;
			q.push(std::make_pair(temp_y, temp_x));
		}
	}
	
	return (0);
}

int		main(void)
{
	int n, m, start_y, start_x;
	int map[1001][1001] = {0,};
	int new_map[1001][1001] = {0,};

	std::cin >> n >> m;
	for(int i = 0; i < n; i++) {
		for(int j = 0; j < m; j++) {
			std::cin >> map[i][j];
			if (map[i][j] == 2) {
				start_y = i;
				start_x = j;
			}
		}
	}

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

	map_scan(map, new_map, n, m, start_y, start_x);

	for(int i = 0; i < n; i++) {
		for(int j = 0; j < m; j++) {
			std::cout << new_map[i][j] << " ";
		}
		std::cout << "\n";
	}
	return (0);
}

코드가 너무 더러운데 리팩토링할 엄두가 안난다..

후기쓰

오랜만에 bfs 관련 문제를 만나보니 많이 당황하고 어려웠던 거 같다.

빠르게 다시 익숙해질 수 있도록 노력해야겠다.