[ 백준 / C++] 3184번 : 양

3184번: 양 (acmicpc.net)

3184번: 양

문제

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.

맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.

아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

 

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.

다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

 

출력

하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.

 

문제설명

dfs나 bfs를 사용해서 문제를 풀 수 있겠네요. 저는 bfs를 사용해서 풀어봤습니다.

한 울타리 안을 검사할 때 'o'와 'v'의 개수를 검사해야 하는데, 한 울타리의 검사를 할 때와 마친 뒤에 변수의 값을 저장해야 하기 때문에 bfs가 실행되는 순서를 생각해 보는 게 어려웠네요. 

 

코드

 

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>

using namespace std;

int dx[4] = { -1, 1, 0, 0 };
int dy[4] = { 0, 0, -1, 1 };

string arr[251][251];
int check[251][251];


int live_sheep = 0, live_wolf = 0;
int ocnt = 0, vcnt = 0;


int main() {
	int i, j, h, w;
	cin >> h >> w;

	string* arr = new string[h];
	for (i = 0; i < h; i++) {
		cin >> arr[i];
		
	}

	for (i = 0; i < h; i++) {
		for (j = 0; j < w; j++) {
			check[i][j] = 0;
		}
	}

	deque<pair<int, int>> q;

	for (i = 0; i < h; i++) {
		for (j = 0; j < w; j++) {
			if (check[i][j] == 0 && arr[i][j] != '#') {
				check[i][j] = 1;
				q.push_back(make_pair(i, j));

				ocnt = 0;
				vcnt = 0;

				if (arr[i][j] == 'v') {
					vcnt++;
				}
				if (arr[i][j] == 'o') {
					ocnt++;
				}

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

					q.pop_front();

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

						if (nx >= 0 && nx < w && ny >= 0 && ny < h) {
							if (check[ny][nx] == 0 && arr[ny][nx] != '#') {
								check[ny][nx] = 1;
								q.push_back(make_pair(ny, nx));

								if (arr[ny][nx] == 'v') {
									vcnt++;
								}
								if (arr[ny][nx] == 'o') {
									ocnt++;
								}

							}

						}
					}
				}

				if (ocnt > vcnt) {
					live_sheep += ocnt;
				}
				else if(ocnt<=vcnt) {
					live_wolf += vcnt;
				}

			}
		}
	}

	printf("%d %d", live_sheep, live_wolf);

	return 0;
}

 

코드설명

기본적으로 사용하는 bfs방식에 ocnt, vcnt, live_sheep. live_wolf 라는 변수를 활용해서 코드를 만들었습니다.

덱에 pair를 추가할 때(i, j나 ny, nx 위치에서 #가 아니며, check이 0인 경우) 조건에 따라 ocnt와 vcnt를 추가하여 한 울타리 안을 검사할 때 양과 늑대의 마리수를 셀 수 있게 했습니다.

또한, while문이 종료될 때(한 울타리 안의 검사를 마쳤을 때) ocnt와 vcnt의 값에 따라 live_sheep과 live_wolf라는 변수에 더해서 최종적으로 결과값을 출력할 수 있게 했습니다.