백준 2206번 C++

백준 2206번 벽 부수고 이동하기

https://www.acmicpc.net/problem/2206

2206번: 벽 부수고 이동하기

정답 코드

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
#include<string>

using namespace std;

int map[1001][1001];
int visited[1001][1001][2];

queue<pair<pair<int,int>,int>> q;
string str;
int n,m;
int dx[] = {1,0,-1,0};
int dy[] = {0,1,0,-1};

int bfs(){
    q.push({{1,1},1});
    visited[1][1][1]=1;

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

        if(x==n && y==m) return visited[x][y][wall];

        for(int i=0; i<4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx >= 1 && nx <= n && ny >= 1 && ny <=m) {
                if(wall == 1 && map[nx][ny]==1){ 
                    q.push({{nx,ny},0});
                    visited[nx][ny][0] = visited[x][y][1] + 1;
                }
                else if(map[nx][ny]==0 && visited[nx][ny][wall]==0){
                    q.push({{nx,ny},wall});
                    visited[nx][ny][wall] = visited[x][y][wall] + 1;
                }
            }
        }
    }
    return -1;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    cin >> n >> m;

    for(int i=1; i<=n;i++){
        cin >> str;
        for(int j=1; j<=str.size();j++){
            map[i][j] = str[j-1]-48;
        }
    }
    int ans = bfs();

    cout << ans;
    return 0;
}

문제 풀이

bfs로 구현하는 문제이지만 이전에 풀었던 문제들보다는 조금 더 난이도가 있는 문제입니다. 그리고 배열을 선언할 시 3차원으로 선언해서 벽 부수는 것까지 고려를 해주어야 합니다.
도착점에 도달할 시 거기에 해당하는 visit배열의 좌표를 반환해줍니다. 그게 아닐 시 for문을 돌리면서 wall=1이고 움직이는 곳이 1일때와
움직이는 곳이0이고  방문하지 않은 경우를 나누어서
구현하면 됩니다. 이 두가지 이외에는 일반적인 bfs구현과 동일합니다.