[BFS] 3184번 양

3184번 양

 

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

 

문제

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

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

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

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

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

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

 

입력

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

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

 

출력

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

 

예제 입력

6 6
...#..
.##v#.
#v.#.#
#.o#.#
.###.#
...###

8 8
.######.
#..o...#
#.####.#
#.#v.#.#
#.#.o#o#
#o.##..#
#.v..v.#
.######.

9 12
.###.#####..
#.oo#...#v#.
#..o#.#.#.#.
#..##o#...#.
#.#v#o###.#.
#..#v#....#.
#...v#v####.
.####.#vv.o#
.......####.

 

예제 출력

0 2

3 1

3 5

 

해결방법

영역을 구해서 조건을 처리하는 간단한 BFS 탐색 문제이다

 

만약 양과 늑대가 둘 다 없는 영역은 탐색을 할 필요가 없다

따라서 노드가 'o' 이거나 'v' 인 경우에 탐색을 수행했다

 

빈칸이거나 늑대 또는 양인 칸의 영역을 구한 다음, 늑대의 수와 양의 수를 카운트한다

그리고 아래의 조건을 처리한다

if(o_cnt > v_cnt) v_cnt = 0;
else o_cnt = 0;

o_ans += o_cnt;
v_ans += v_cnt;

 

 

소스코드

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <bitset>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#define MAX 251
#define INF 987654321
using namespace std;

typedef pair<int, int> pii;

int n,m;
int o_ans,v_ans;

char map[MAX][MAX];
bool chk[MAX][MAX];

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

void bfs(int r,int c){
    int o_cnt = 0;
    int v_cnt = 0;
    
    queue<pii> q;
    q.push(pii(r,c));
    chk[r][c] = true;
    
    if(map[r][c]=='o') o_cnt++;
    else if(map[r][c]=='v') v_cnt++;
    
    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>=n || ny<0 || ny>=m) continue;
            
            if(!chk[nx][ny] && map[nx][ny] != '#'){
                chk[nx][ny] = true;
                q.push(pii(nx,ny));
                
                if(map[nx][ny] == 'o') o_cnt++;
                else if(map[nx][ny] == 'v') v_cnt++;
            }
        }
    }
    if(o_cnt > v_cnt) v_cnt = 0;
    else o_cnt = 0;
    
    o_ans += o_cnt;
    v_ans += v_cnt;
}

int main(int argc, const char * argv[]) {
    // cin,cout 속도향상
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    
    cin >> n >> m;
    
    memset(chk, false, sizeof(chk));
    
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            cin >> map[i][j];
        }
    }
    
    o_ans = 0;
    v_ans = 0;
    
    // BFS
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            if((map[i][j] == 'v' || map[i][j] == 'o') && !chk[i][j]){
                bfs(i,j);
            }
        }
    }
    
    cout << o_ans << " " << v_ans << "\n";
}