[BOJ 27211 // C++] 도넛 행성

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 27211번 문제인 도넛 행성이다.
문제는 아래 링크를 확인하자.

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

27211번: 도넛 행성

주어진 도넛 행성 위에서의 움직임은 N행 M열로 구성된 격자판 위에서 N행의 다음 행은 1행이고 M열의 다음 열은 1열과 같은 움직임으로 나타낼 수 있음을 관찰하자.

 

위와 같은 관찰을 토대로, 주어진 N행M열의 배치를 숲이 아닌 칸을 노드로, 도넛 행성 위에서의 서로 인접한 숲이 아닌 칸의 관계를 에지로 하는 그래프로 생각하자. 이 때, 문제에서 구하고자 하는 값은 해당 그래프의 connected component의 개수와 같아지므로 그 개수를 세는 것으로 문제를 해결할 수 있다.

 

아래는 제출한 소스코드이다.

#include <iostream>
#include <queue>
#include <vector>
using namespace std;

int R, C;
int board[1000][1000];
int ans;
int dr[4] = { 1,-1,0,0 };
int dc[4] = { 0,0,1,-1 };

void bfs(int sr, int sc) {
	board[sr][sc] = 1;
	queue<pair<int, int>> que;
	que.push(make_pair(sr, sc));
	while (!que.empty()) {
		int r = que.front().first, c = que.front().second; que.pop();
		for (int i = 0; i < 4; i++) {
			int rr = r + dr[i], cc = c + dc[i];
			if (rr < 0) rr = R - 1;
			else if (rr == R) rr = 0;
			if (cc < 0) cc = C - 1;
			else if (cc == C) cc = 0;

			if (!board[rr][cc]) {
				board[rr][cc] = 1;
				que.push(make_pair(rr, cc));
			}
		}
	}
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> R >> C;
	for (int r = 0; r < R; r++) {
		for (int c = 0; c < C; c++) {
			cin >> board[r][c];
		}
	}

	for (int r = 0; r < R; r++) {
		for (int c = 0; c < C; c++) {
			if (board[r][c]) continue;
			bfs(r, c);
			ans++;
		}
	}

	cout << ans;
}