All Articles

[BOJ 3184] 양

문제

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

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

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

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

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

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

코드

#include <iostream>
#include <vector>
#include <map>

using namespace std;

int r, c;
int dir_x[] = { 1, -1, 0, 0 }, dir_y[] = { 0, 0, 1, -1 };
map<int, int> wolf, sheep;

void dfs(vector<vector<char>>& v, vector<vector<bool>> &is_visited, int x, int y, int team) {
  is_visited[x][y] = true;
  if (v[x][y] == 'o') {
    sheep[team]++;
  } else if (v[x][y] == 'v') {
    wolf[team]++;
  }

  for (int i = 0; i < 4; i++) {
    int newx = x + dir_x[i], newy = y + dir_y[i];

    if (newx < 0 || newy < 0 || newx >= r || newy >= c) continue;

    if (!is_visited[newx][newy] && v[newx][newy] != '#') {
      dfs(v, is_visited, newx, newy, team);
    }
  }
}

int main() {
  cin >> r >> c;

  vector<vector<char>> v(r, vector<char>(c));
  for (int i = 0; i < r; i++) {
    string tmp;
    cin >> tmp;
    for (int j = 0; j < c; j++) v[i][j] = tmp[j];
  }

  vector<vector<bool>> is_visited(r, vector<bool>(c, false));
  int team = 1;
  for (int i = 0; i < r; i++) {
    for (int j = 0; j < c; j++) {
      if (!is_visited[i][j] && v[i][j] != '#') {
        dfs(v, is_visited, i, j, team++);
      }
    }
  }

  int num_w = 0, num_o = 0;
  for (int i = 1; i < team; i++) {
    if (sheep[i] > wolf[i]) num_o += sheep[i];
    else num_w += wolf[i];
  }
  cout << num_o << " " << num_w;
}

풀이

각 점에 대해서 DFS를 수행하는데, #(벽)이 아니면 모두 방문한다. 이때, 양과 늑대 수를 세면서 주위에 방문 가능할 시 같은 팀으로 표시한다. 방문이 모두 끝나면 map에 들어있는 팀마다 양과 늑대 수를 토대로 결과값을 계산한다.