문제
알고스팟 운영진이 모두 미로에 갇혔다. 미로는 N*M 크기이며, 총 1*1크기의 방으로 이루어져 있다. 미로는 빈 방 또는 벽으로 이루어져 있고, 빈 방은 자유롭게 다닐 수 있지만, 벽은 부수지 않으면 이동할 수 없다.
알고스팟 운영진은 여러명이지만, 항상 모두 같은 방에 있어야 한다. 즉, 여러 명이 다른 방에 있을 수는 없다. 어떤 방에서 이동할 수 있는 방은 상하좌우로 인접한 빈 방이다. 즉, 현재 운영진이 (x, y)에 있을 때, 이동할 수 있는 방은 (x+1, y), (x, y+1), (x-1, y), (x, y-1) 이다. 단, 미로의 밖으로 이동 할 수는 없다.
벽은 평소에는 이동할 수 없지만, 알고스팟의 무기 AOJ를 이용해 벽을 부수어 버릴 수 있다. 벽을 부수면, 빈 방과 동일한 방으로 변한다.
만약 이 문제가 알고스팟에 있다면, 운영진들은 궁극의 무기 sudo를 이용해 벽을 한 번에 다 없애버릴 수 있지만, 안타깝게도 이 문제는 Baekjoon Online Judge에 수록되어 있기 때문에, sudo를 사용할 수 없다.
현재 (1, 1)에 있는 알고스팟 운영진이 (N, M)으로 이동하려면 벽을 최소 몇 개 부수어야 하는지 구하는 프로그램을 작성하시오.
풀이
다익스트라 알고리즘을 활용하였다.
본래 최단거리를 저장할 1차배열을 그래프와 같은 이차배열로 선언하여,
(x+1, y), (x, y+1), (x-1, y), (x, y-1) 네가지 방향으로 검색을 하여 찾았다.
(문제에서는 x 가 n , y 가 m 인데 이차배열 특성상 반대로 보이는게 불편해서 편의상 x = m , y = n 으로 풀었다.)
▶ x = m 이라면,
(x + 1) 이 불가능하다.
▶ y = n 이라면,
(y + 1)이 불가능하다.
▶ x = 1 이라면,
(x - 1)이 불가능하다.
▶ y = 1 이라면,
(y - 1)이 불가능하다.
이렇게 4가지 경우를 제외하고 나머지들은 현재의 위치에서 다음위치의 값을 더한값이 다음위치의 값보다 크면 바꿔주면된다.
if( map[r.x + 1][r.y] > r.c + graph[r.x+1][r.y]) {
map[r.x + 1][r.y] = r.c + graph[r.x+1][r.y];
pq.add(new room(r.x +1 ,r.y, map[r.x+1][r.y]));
}
위와 같이 변경하고 우선순위 큐에 넣어주었다.
<전체코드>
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
public class Main {
static StringBuilder sb = new StringBuilder();
static int[][] map, graph;
static int n, m;
static class room implements Comparable<room>{
int x, y, c;
public room(int x, int y, int c) {
this.x = x;
this.y = y;
this.c = c;
}
@Override
public int compareTo(room p1) {
// TODO Auto-generated method stub
return Integer.compare(this.c, p1.c);
}
}
public static void main(String[] args) throws IOException{
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
String[] nm = bufferedReader.readLine().split("\\s");
n = Integer.parseInt(nm[0]);
m = Integer.parseInt(nm[1]);
graph = new int[m+1][n+1];
map = new int[m+1][n+1];
for(int i = 1 ; i <= m; i++) {
String[] number = bufferedReader.readLine().split("");
for(int j = 1 ; j <= n ; j++) {
graph[i][j] = Integer.parseInt(number[j-1]);
map[i][j] = Integer.MAX_VALUE;
}
}
map[1][1] = 0;
findRoute(1, 1);
System.out.println(map[m][n]);
}
static void findRoute(int x, int y) {
PriorityQueue<room> pq = new PriorityQueue<room>();
pq.add(new room(1, 1, 0));
while(!pq.isEmpty()) {
room r = pq.poll();
if(r.x < m && map[r.x][r.y] != Integer.MAX_VALUE) {
if( map[r.x + 1][r.y] > r.c + graph[r.x+1][r.y]) {
map[r.x + 1][r.y] = r.c + graph[r.x+1][r.y];
pq.add(new room(r.x +1 ,r.y, map[r.x+1][r.y]));
}
}
if(r.x > 1 && map[r.x][r.y] != Integer.MAX_VALUE) {
if( map[r.x][r.y - 1] > r.c + graph[r.x][r.y -1]) {
map[r.x][r.y - 1] = r.c + graph[r.x][r.y - 1];
pq.add(new room(r.x,r.y - 1, map[r.x][r.y -1]));
}
}
if(r.y < n && map[r.x][r.y] != Integer.MAX_VALUE) {
if( map[r.x][r.y + 1] > r.c + graph[r.x][r.y + 1]) {
map[r.x][r.y + 1] = r.c + graph[r.x][r.y + 1];
pq.add(new room(r.x ,r.y + 1, map[r.x][r.y + 1]));
}
}
if(r.y > 1 && map[r.x][r.y] != Integer.MAX_VALUE) {
if( map[r.x - 1][r.y] > r.c + graph[r.x - 1][r.y]) {
map[r.x - 1][r.y] = r.c + graph[r.x - 1][r.y];
pq.add(new room(r.x -1 ,r.y, map[r.x - 1][r.y]));
}
}
}
}
}
'알고리즘 공부 > 최단거리 알고리즘' 카테고리의 다른 글
백준 1922 (네트워크)* - 크루스칼알고리즘(Union-Find 사용) (0) | 2021.04.27 |
---|---|
백준 1761 (정점들의 거리) - LCA 풀이 (0) | 2021.03.29 |
백준 1238 (파티) / 다익스트라 (0) | 2021.02.03 |
백준 5719 (거의 최단경로) | 다익스트라 , BFS (0) | 2021.01.28 |
백준 10282 (해킹) | 다익스트라 알고리즘(Dijkstra Algorithm) (0) | 2021.01.27 |