알고리즘 공부/최단거리 알고리즘

백준 1261(알고스팟) / 다익스트라

kdhoooon 2021. 1. 30. 22:47

문제


알고스팟 운영진이 모두 미로에 갇혔다. 미로는 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]));
				}
			}
		}
		
		
	}
}