[BOJ / C++] 3187번 : 양치기 꿍

문제

양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.

하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.

꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?

단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

입력

입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.

다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.

출력

살아남게 되는 양과 늑대의 수를 각각 순서대로 출력한다.

Solved.ac 레벨

실버 I

풀이

#include <bits/stdc++.h>
#define MAX 251


using namespace std;

int r, c, kCnt, vCnt;
int dx[4] = {0, 0, -1, 1};
int dy[4] = {-1, 1, 0, 0};
char node[MAX][MAX];
bool visited[MAX][MAX];

queue<pair<int, int> > q;

void BFS(pair<int, int> p) {
    int curK = 0, curV = 0;
    q.push(p);
    visited[p.first][p.second] = true;

    if(node[p.first][p.second] == 'k') curK++;
    if(node[p.first][p.second] == 'v') curV++;

    while(!q.empty()) {
        int curX = q.front().first;
        int curY = q.front().second;
        q.pop();

        for(int i = 0; i < 4; i++) {
            int nextX = curX + dx[i];
            int nextY = curY + dy[i];

            if(nextX >= 0 && nextY >= 0 && nextX < r && nextY <c) {
                if(!visited[nextX][nextY]) {
                    if(node[nextX][nextY] == '.') {
                        //다음 노드가 이동 가능한 공간일 경우
                        visited[nextX][nextY] = true;
                        q.push(make_pair(nextX, nextY));
                    } else if(node[nextX][nextY] == 'k') {
                        //다음 노드에 양이 있을 경우
                        curK++;
                        visited[nextX][nextY] = true;
                        q.push(make_pair(nextX, nextY));
                    } else if(node[nextX][nextY] == 'v') {
                        //다음 노드에 늑대가 있을 경우
                        curV++;
                        visited[nextX][nextY] = true;
                        q.push(make_pair(nextX, nextY));
                    }
                }
            }
        }

    }
    //양이 늑대보다 많은 경우
    if(curK > curV) kCnt += curK;
    else vCnt += curV;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> r >> c;

    for(int i = 0; i < r; i++) {
        for(int j = 0; j < c; j++) {
            cin >> node[i][j];
        }
    }

    for(int i = 0; i < r; i++) {
        for(int j = 0; j < c; j++) {
            if(node[i][j] == '#') continue;
            if(!visited[i][j]) {
                BFS(make_pair(i, j));
            }
        }
    }

    cout << kCnt << " " << vCnt << "\n";

    return 0;
}

 

3187번: 양치기 꿍