백준 3187: 양치기 꿍 [Java] - 포포

문제

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

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

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

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

입력

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

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

출력

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

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

3187번: 양치기 꿍

---

public class No3187 {
    static char[][] arr;
    static boolean[][] visit;
    static int n;
    static int m;
    static int[] dx = {1, 0, -1, 0};
    static int[] dy = {0, 1, 0, -1};
    static Queue<Point> qu;
    static int sheep;
    static int wolf;
    static int sheepAns;
    static int wolfAns;
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringBuffer sb = new StringBuffer();
        n = Integer.parseInt(st.nextToken(" "));
        m = Integer.parseInt(st.nextToken(" "));
        arr = new char[n][m];
        visit = new boolean[n][m];
		
        // 1. 배열에 입력 요소들을 담는다.
        for (int i = 0; i < n; i++) {
            String s = br.readLine();
            for (int j = 0; j < m; j++) {
                arr[i][j] = s.charAt(j);
            }
        }

        qu = new LinkedList<>();
		
        // 2. 배열을 돌면서 방문하지 않은 양 or 늑대를 만나는 경우 bfs를 실행한다.
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if((arr[i][j] == 'k' || arr[i][j] == 'v') && !visit[i][j]){
                    bfs(i, j);
                }
            }
        }
		// 결과 답 출력
        System.out.println(sheepAns+" "+ wolfAns);
    }

    private static void bfs(int x, int y) {
        sheep = 0;
        wolf = 0;
        if(arr[x][y] == 'v'){
            wolf++;
        }else{
            sheep++;
        }

        visit[x][y] = true;
        qu.offer(new Point(x,y));

        while (!qu.isEmpty()) {
            Point p = qu.poll();

            for (int i = 0; i < 4; i++) {
                int nx = p.x + dx[i];
                int ny = p.y + dy[i];

                if(nx < 0 || ny< 0 || nx>= n || ny >= m) continue;

                if(!visit[nx][ny] && arr[nx][ny] != '#'){
                    qu.offer(new Point(nx, ny));
                    visit[nx][ny] = true;

                    if(arr[nx][ny] == 'v'){
                        wolf++;
                    }else if(arr[nx][ny] == 'k'){
                        sheep++;
                    }
                }
            }
        }
        // 양보다 늑대의 수가 많거나 같으면, 늑대만 더해준다.
        // 양이 더 많은 경우 양만 더해준다.
        if(sheep <= wolf) {
            wolfAns += wolf;
        }
        if(sheep > wolf){
            sheepAns += sheep;
        }
    }
    static class Point{
        int x;
        int y;

        public Point(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }
}

주석에 설명을 간단히 달아놓았다.

1. 배열에 입력 요소들을 저장한다.

2. 배열을 돌면서 방문하지 않은 양 or 늑대를 만나는 경우 bfs를 실행한다!

3. bfs를 돌고, Queue가 비어 bfs가 종료되는 시점에 양의 수, 늑대의 수를 비교하여 더 많은 쪽만 더해준다.

4. 결과 출력!