백준 1261 알고스팟 (C++)

1261번: 알고스팟

이 문제는 다익스트라나 0-1 BFS로 풀 수 있는 문제이다. 나는 다익스트라로 풀어봤다. d 배열은 현재 좌표까지 벽을 부순 최솟값을 나타낸다. 입력받은 맵을 bfs를 돌면서 현재 좌표와 다음 이동할 좌표의 d 배열값을 비교하고 최솟값을 저장해주었다. 이를 반복하면 d[N-1][M-1] 값을 구할 수 있게되고 이를 출력해주었다. 어렵지 않게 풀 수 있었던 문제였다. 0-1 BFS 유형도 연습을 해봐야겠다.

#include <iostream>
#include <algorithm>
#include <queue>

#define INF 987654321

using namespace std;

int N, M;
string A[100];
int d[100][100];
int dy[4] = { -1,1,0,0 };
int dx[4] = { 0,0,-1,1 };

void bfs() {
    queue<pair<int, int>> q;

    d[0][0] = 0;
    q.push({0,0});

    while (!q.empty()) {
        int y = q.front().first;
        int x = q.front().second;

        q.pop();

        for (int i = 0; i < 4; i++) {
            int ny = y + dy[i];
            int nx = x + dx[i];

            if (ny < 0 || nx < 0 || ny >= N || nx >= M) continue;

            if (A[ny][nx] == '1') {
                if (d[ny][nx] <= d[y][x] + 1) continue;

                d[ny][nx] = d[y][x] + 1;
            }
            else {
                if (d[ny][nx] <= d[y][x]) continue;

                d[ny][nx] = d[y][x];
            }

            q.push({ ny,nx });
        }
    }
}

void solution() {
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < M; j++) {
            d[i][j] = INF;
        }
    }

    bfs();

    cout << d[N - 1][M - 1];
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    cin >> M >> N;

    for (int i = 0; i < N; i++) {
        cin >> A[i];
    }

    solution();

    return 0;
}