본문 바로가기
백준 알고리즘/BFS

백준 2206번 C++

by paysmile 2019. 8. 15.

#include <iostream>
#include <string>
#include <queue>

using namespace std;
const int MAX = 1000;
int map[MAX][MAX];
int n, m;
int visited[MAX][MAX][2];

struct Move {
	int x, y;
};
Move mv[4] = { {1,0}, {-1,0} ,{0,1}, {0,-1} };

int bfs() {
	queue<pair<int, pair<int,int>>> q;
	q.push(make_pair(0,make_pair(0,1)));
	visited[0][0][1] = 1;

	while (!q.empty()) {
		int currenti = q.front().first;
		int currentj = q.front().second.first;
		int broke = q.front().second.second;
		q.pop();

		if (currenti == (n - 1) && currentj == (m - 1))
			return visited[currenti][currentj][broke];

		for (int k = 0; k < 4; k++) {
			int movei = currenti + mv[k].x;
			int movej = currentj + mv[k].y;

			if (movei >= 0 && movei < n && movej >= 0 && movej < m) {
				if (map[movei][movej] == 1 && broke == 1) {
					q.push(make_pair(movei, make_pair(movej, broke - 1)));
					visited[movei][movej][broke - 1] = visited[currenti][currentj][broke] + 1;
				}

				if (map[movei][movej] == 0 && visited[movei][movej][broke] == 0) {
					q.push(make_pair(movei, make_pair(movej, broke)));
					visited[movei][movej][broke] = visited[currenti][currentj][broke] + 1;
				}
			}
		}
	}
	return -1;
}

int main(void) {
	string s;
	cin >> n >> m;
	
	for (int i = 0; i < n; i++) {
		cin >> s;
		for (int j = 0; j < m; j++) {
			map[i][j] = s[j] - '0';
		}
	}
	int answer = bfs();
	cout << answer << endl;
	return 0;
}

'백준 알고리즘 > BFS' 카테고리의 다른 글

백준 1963번 C++  (0) 2019.08.16
백준 2573번 C++  (0) 2019.08.15
백준 3055번 C++  (0) 2019.08.14
백준 1707번 C++  (0) 2019.08.14
백준 5014번 C++  (0) 2019.08.12