[C/C++] 백준 - 3184번 (양)

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

3184번: 양

문제

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

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

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

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

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

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

입력

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

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

출력

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

 

DFS를 이용하여 쉽게 해결할 수 있는 문제이다.

양과 늑대를 범위 내에서 세어 양의 수가 더 많으면 양의 수만 카운트 해주고, 그렇지 않은 경우는 늑대만 카운트 해주면 된다.

#include <iostream>
#include <queue>
#include <vector>
#include <stack>
#include <string>
#include <cstring>
#include <sstream>
#include <set>
#include <unordered_set>
#include <map> 
#include <algorithm>
#include <cmath>
#include <ctime>
#define fastio ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ENDL cout << endl
#define ll long long
#define ull unsigned long long
#define INF 18549876543
#define Mod 1000000009
#define endl '\n'
#define pil pair<int,int>

using namespace std;

int N, M, s, w;
// 양은 1, 늑대는 2
int board[253][253];
bool visited[253][253];

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

bool isValid(int y, int x) {
	return (0 <= y && y < N && 0 <= x && x < M);
}

void DFS(int y, int x) {
	pair<int, int> num;
	visited[y][x] = true;

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

		if (isValid(ny, nx)) {
			if (!visited[ny][nx] && board[ny][nx] != -1) {
				if (board[ny][nx] == 0) DFS(ny, nx);
				else if (board[ny][nx] == 1) s++, DFS(ny, nx);
				else if (board[ny][nx] == 2) w++, DFS(ny, nx);
			}
		}
	}
}

int main() {
	fastio;
	cin >> N >> M;
	for (int i = 0; i < N; i++) {
		string s;
		cin >> s;
		for (int j = 0; j < M; j++) {
			if (s[j] == '.') board[i][j] = 0;
			else if (s[j] == '#') board[i][j] = -1;
			else if (s[j] == 'o') board[i][j] = 1;
			else if (s[j] == 'v') board[i][j] = 2;
		}
	}

	pair<int, int> answer(0, 0);
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (!visited[i][j] && board[i][j] != -1) {
				s = 0, w = 0;
				if (board[i][j] == 0) {
					DFS(i, j);
					if (s > w) answer.first += s;
					else answer.second += w;
				}
				else if (board[i][j] == 1) {
					s = 1;
					DFS(i, j);
					if (s > w) answer.first += s;
					else answer.second += w;
				}
				else {
					w = 1;
					DFS(i, j);
					if (s > w) answer.first += s;
					else answer.second += w;
				}
			}
		}
	}
	cout << answer.first << " " << answer.second << endl;
	 
	return 0;
}