백준 10026번 C++

백준 10026번 적록색약

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

10026번: 적록색약

정답 코드

#include<iostream>
#include<queue>
using namespace std;
char arr[101][101];
int vis[101][101];
int dx[4] = { 1,0,-1,0 };
int dy[4] = { 0,1,0,-1 };
int four(int a) {
	queue<pair<int, int>>q;
	int num = 0;
	for (int x = 0; x < a; x++) {
		for (int y = 0; y < a; y++) {
			if (vis[x][y])
				continue;
			if (arr[x][y] == 'R') {
				num++;
				q.push({ x,y });
				while (!q.empty()) {
					pair<int, int>cur = q.front();
					q.pop();
					for (int x = 0; x < 4; x++) {
						int mx = cur.first + dx[x];
						int my = cur.second + dy[x];
						if (mx < 0 || my < 0 || mx >= a || my >= a)
							continue;
						if (arr[mx][my] != 'R' || vis[mx][my])
							continue;
						vis[mx][my] = 1;
						q.push({ mx,my });

					}
				}

			}
			if (arr[x][y] == 'G') {
				num++;
				q.push({ x,y });
				while (!q.empty()) {
					pair<int, int>cur = q.front();
					q.pop();
					for (int x = 0; x < 4; x++) {
						int mx = cur.first + dx[x];
						int my = cur.second + dy[x];
						if (mx < 0 || my < 0 || mx >= a || my >= a)
							continue;
						if (arr[mx][my] != 'G' || vis[mx][my])
							continue;
						vis[mx][my] = 1;
						q.push({ mx,my });

					}
				}

			}
			if (arr[x][y] == 'B') {
				num++;
				q.push({ x,y });
				while (!q.empty()) {
					pair<int, int>cur = q.front();
					q.pop();
					for (int x = 0; x < 4; x++) {
						int mx = cur.first + dx[x];
						int my = cur.second + dy[x];
						if (mx < 0 || my < 0 || mx >= a || my >= a)
							continue;
						if (arr[mx][my] != 'B' || vis[mx][my])
							continue;
						vis[mx][my] = 1;
						q.push({ mx,my });

					}
				}

			}

		}
	}

	return num;
	

}
int three(int a) {
	for (int x = 0; x < a; x++) {
		for (int y = 0; y < a; y++)
			vis[x][y] = 0;
	}
	queue<pair<int, int>>q;
	int num = 0;
	for (int x = 0; x < a; x++) {
		for (int y = 0; y < a; y++) {
			if (vis[x][y])
				continue;
			if (arr[x][y] == 'B') {
				num++;
				q.push({ x,y });
				while (!q.empty()) {
					pair<int, int>cur = q.front();
					q.pop();
					for (int x = 0; x < 4; x++) {
						int mx = cur.first + dx[x];
						int my = cur.second + dy[x];
						if (mx < 0 || my < 0 || mx >= a || my >= a)
							continue;
						if (arr[mx][my] != 'B' || vis[mx][my])
							continue;
						vis[mx][my] = 1;
						q.push({ mx,my });

					}
				}

			}
			else {
				num++;
				q.push({ x,y });
				while (!q.empty()) {
					pair<int, int>cur = q.front();
					q.pop();
					for (int x = 0; x < 4; x++) {
						int mx = cur.first + dx[x];
						int my = cur.second + dy[x];
						if (mx < 0 || my < 0 || mx >= a || my >= a)
							continue;
						if (arr[mx][my] =='B' || vis[mx][my])
							continue;
						vis[mx][my] = 1;
						q.push({ mx,my });

					}
				}

			}
		}
	}
	return num;
}
int main() {
	int a;
	cin >> a;
	for (int x = 0; x < a; x++) {
		for (int y = 0; y < a; y++) {
			cin >> arr[x][y];
		}
	}
	cout << four(a) << " ";
	cout<< three(a);
}

문제 풀이

bfs를 이용해서 문제를 풀었습니다. 적록색약 구역을 구하는 문제인데 적록색약인 사람이 볼 수 있는 경우와
적록색약이 아닌 사람이 볼 수 있는 경우의 수를 나누어서 풀었습니다. 전자는 r,g를 같은 선상으로 취급하고
후자는 r,g를 다른 선상으로 취급했습니다. 배열의.범위를 벗어나거나 이미 방문한 곳들은 continue시켜서 문제를 해결했습니다.