[백준] 3184번. 양

[백준] 3184번. 양

1. 문제

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

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

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

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

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

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

2. 입력

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

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

3. 출력

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

4. 풀이

• BFS로 풀었다.
• 영역마다 BFS를 하였고 영역 안의 양의 수와 늑대의 수를 구하고 영역의 BFS가 끝날 때 양의 수가 더 많으면 총 양의 수에 더해주고 반대면 총 늑대의 수를 더해줬다.
  더 작은 수는 잡아먹히기 때문에 무시한다.
• 울타리가 없는 곳은 어느 영역도 아니고 늑대와 양은 항상 영역에 있기 때문에 신경쓰지 않아도 된다.

5. 처음 코드와 달라진 점

• 방문처리 안해줌...ㅎ

6. 코드

#include <iostream>
#include <queue>

using namespace std;

char map[250][250];
bool visited[250][250];
int r, c;
int dx[] = { 0,1,0,-1 };
int dy[] = { 1,0,-1,0 };
int total_number_of_lambs = 0;
int total_number_of_wolves = 0;

void bfs(int start_x, int start_y) {
    int number_of_wolves=0;
    int number_of_lambs=0;
    if (map[start_x][start_y] == 'o') number_of_lambs++;
    if (map[start_x][start_y] == 'v') number_of_wolves++;
    visited[start_x][start_y] = true;

    queue<pair<int, int>> q;
    q.push({ start_x, start_y });
    while (!q.empty()){
        int x = q.front().first;
        int y = q.front().second;
        q.pop();
        for (int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];

            if (nx < 0 || nx >= r || ny < 0 || ny >= c) continue;
            if (map[nx][ny] == '#' || visited[nx][ny]) continue;

            if (map[nx][ny] == 'o') number_of_lambs++;
            if (map[nx][ny] == 'v') number_of_wolves++;
            visited[nx][ny] = true;
            q.push({ nx,ny });
        }
    }
    if (number_of_lambs > number_of_wolves) total_number_of_lambs += number_of_lambs;
    else total_number_of_wolves += number_of_wolves;
}

int main(void) {
    cin.tie(NULL);
    ios_base::sync_with_stdio(false);
    
    cin >> r >> c;
    for (int i = 0; i < r; i++)
        for (int j = 0; j < c; j++)
            cin >> map[i][j];
    
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            if (!visited[i][j] && map[i][j] != '#') bfs(i, j);
        }
    }
    cout << total_number_of_lambs << " " << total_number_of_wolves;
        
}