[백준] 3184번 양

3184 양

문제

---

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

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

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

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

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

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

입력

---

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

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

출력

---

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

---

풀이)

dfs를 이용해서 쉽게 푼 문제이다.

과정은 간단하다.

1. 울타리 안에 있는 영역을 구한다. (dfs를 통해 칸이 울타리가 아닌 경우는 계속 탐색해주었다.)

2. 그 영역안에서 양의 수와 늑대의 수를 구한다.

3. 영역 탐색을 마친 후, 양이 늑대보다 많다면 양을 살리고 그렇지 않다면 늑대를 살려 놓았다.

위의 1,2,3을 전체 배열에서 반복하면된다. 주의할 점은 bool visited[][]를 만들어 놓아야

중복되는 위치는 확인을 안한다는 것이다.

코드)

#include <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
 
int r, c;
char map[250][250];
int dy[4] = { -1,1,0,0 };
int dx[4] = { 0,0,-1,1 };
bool visited[250][250];
int wolf, sheep;
 
void dfs(int y,int x) {
    visited[y][x] = true;
    if (map[y][x] == 'v')wolf++;
    else if (map[y][x] == 'o')sheep++;
 
    for (int i = 0; i < 4; i++) {
        int ny = y + dy[i];
        int nx = x + dx[i];
        if (ny >= 0 && ny < r && nx >= 0 && nx < c) {
            if (visited[ny][nx] == true || map[ny][nx] == '#') continue;
            dfs(ny, nx);
        }
    }
}
 
int main() {
    scanf("%d %d", &r, &c);
    getchar();
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            scanf("%c", &map[i][j]);
        }    
        getchar();
    }
    int tv=0, to=0;
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            if (map[i][j] == '.' || map[i][j] == 'v' || map[i][j] == 'o') {
                if (visited[i][j]) continue;
                dfs(i, j);
                if (sheep > wolf) {
                    to += sheep;
                }
                else if (sheep <= wolf) {
                    tv += wolf;
                }
                //다음 탐색을 위해 초기화
                sheep = 0;
                wolf = 0;
            }
        }
    }
    printf("%d %d", to, tv);
    return 0;
}
 

결과)