알고리즘/PS - 백준

[백준 1261 - C++] 알고스팟 : 다익스트라(Dijkstra)

excited-hyun 2021. 2. 3. 16:27
반응형

www.acmicpc.net/problem/1261

 

1261번: 알고스팟

첫째 줄에 미로의 크기를 나타내는 가로 크기 M, 세로 크기 N (1 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 미로의 상태를 나타내는 숫자 0과 1이 주어진다. 0은 빈 방을 의미하고, 1은 벽을 의미

www.acmicpc.net

문제

알고스팟 운영진이 모두 미로에 갇혔다. 미로는 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)으로 이동하려면 벽을 최소 몇 개 부수어야 하는지 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 미로의 크기를 나타내는 가로 크기 M, 세로 크기 N (1 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 미로의 상태를 나타내는 숫자 0과 1이 주어진다. 0은 빈 방을 의미하고, 1은 벽을 의미한다.

(1, 1)과 (N, M)은 항상 뚫려있다.

 

출력

첫째 줄에 알고스팟 운영진이 (N, M)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

 

 

풀이

벽을 부수고 지나갈 때의 비용은 1, 빈방을 지나갈 때의 비용은 0으로 계산하여 다익스트라 알고리즘을 실행하면 된다.

처음에 m이 가로, n이 세로임을 제대로 읽지않고 문제를 풀어 엉뚱한 답이 나와 아래 코드는 입력 받을 때의 n, m의 순서만 고친것이다.

전부 고치기는 너무 귀찮았다ㅠ

#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>

using namespace std;

int map[100][100];
int cost[100][100];

int X[4] = {1, -1, 0, 0};
int Y[4] = {0, 0, 1, -1};

priority_queue<pair<int, pair<int, int>>> pq; //cost, x, y

int main(void){
    int n, m;
    int x, y, cnt;
    int next_x, next_y, next_cnt;
    
    scanf("%d %d", &m, &n);
    
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            scanf("%1d", &map[i][j]);
            cost[i][j] = -1;
        }
    }
    cost[0][0] = 0;
    pq.push(make_pair(0,make_pair(0, 0)));
    
    while(!pq.empty()){
        cnt = -pq.top().first;
        x = pq.top().second.first;
        y = pq.top().second.second;
        pq.pop();
        
        if(cost[x][y] < cnt)
            continue;
        
        for(int i=0; i<4; i++){
            next_x = x + X[i];
            next_y = y + Y[i];
            next_cnt = cnt + map[next_x][next_y];
            
            if(next_x < 0 || next_x >= n || next_y<0 || next_y >= m)
                continue;
            
            if(cost[next_x][next_y] == -1){
                cost[next_x][next_y] = next_cnt;
                pq.push(make_pair(-next_cnt,make_pair(next_x, next_y)));
            }
            else if(cost[next_x][next_y] > next_cnt){
                cost[next_x][next_y] = next_cnt;
                pq.push(make_pair(-next_cnt,make_pair(next_x, next_y)));
            }
        }
    }
    printf("%d\n", cost[n-1][m-1]);
}
728x90
반응형