[백준] 11559 - Puyo Puyo (java)

Baekjoon 11559- Puyo Puyo (클릭 시 이동)

문제

뿌요뿌요의 룰은 다음과 같다.

필드에 여러 가지 색깔의 뿌요를 놓는다. 뿌요는 중력의 영향을 받아 아래에 바닥이나 다른 뿌요가 나올 때까지 아래로 떨어진다.

• 뿌요를 놓고 난 후, 같은 색 뿌요가 4개 이상 상하좌우로 연결되어 있으면 연결된 같은 색 뿌요들이 한꺼번에 없어진다. 이때 1연쇄가 시작된다.

• 뿌요들이 없어지고 나서 위에 다른 뿌요들이 있다면, 역시 중력의 영향을 받아 차례대로 아래로 떨어지게 된다.

• 아래로 떨어지고 나서 다시 같은 색의 뿌요들이 4개 이상 모이게 되면 또 터지게 되는데, 터진 후 뿌요들이 내려오고 다시 터짐을 반복할 때마다 1연쇄씩 늘어난다.

• 터질 수 있는 뿌요가 여러 그룹이 있다면 동시에 터져야 하고 여러 그룹이 터지더라도 한번의 연쇄가 추가된다.

남규는 최근 뿌요뿌요 게임에 푹 빠졌다. 이 게임은 1:1로 붙는 대전게임이라 잘 쌓는 것도 중요하지만, 상대방이 터뜨린다면 연쇄가 몇 번이 될지 바로 파악할 수 있는 능력도 필요하다. 하지만 아직 실력이 부족하여 남규는 자기 필드에만 신경 쓰기 바쁘다. 상대방의 필드가 주어졌을 때, 연쇄가 몇 번 연속으로 일어날지 계산하여 남규를 도와주자!

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
1 초	256 MB	15087	5752	4114	36.680%

입력

총 12개의 줄에 필드의 정보가 주어지며, 각 줄에는 6개의 문자가 있다.

이때 .은 빈공간이고 .이 아닌것은 각각의 색깔의 뿌요를 나타낸다.

R은 빨강, G는 초록, B는 파랑, P는 보라, Y는 노랑이다.

입력으로 주어지는 필드는 뿌요들이 전부 아래로 떨어진 뒤의 상태이다. 즉, 뿌요 아래에 빈 칸이 있는 경우는 없다.

출력

현재 주어진 상황에서 몇연쇄가 되는지 출력한다. 하나도 터지지 않는다면 0을 출력한다.

풀이

1. 주어진 입력을 순차적으로 탐색하며 '.' 이 아닌 값이 나오면 사방탐색을 통해 방문 같은 색을 방문하고 좌표를 List에 저장한다.
2. List의 size가 4이상이면 List에 저장된 좌표들을 순회하며 '.'으로 바꿔준다.
3. 탐색한 위치는 방문체크를 한 뒤 방문하지 않은곳이 없을 때까지 진행한다(완전 탐색)
4. 2의 과정을 수행했다면 true를 리턴하고 다음 연쇄반응을 일으킬 수 있게 count를 증가시킨다.(count가 0이면 종료)
5. 중력을 작용하고 1번과정을 다시 진행한다.

과정

1. 탐색

   static boolean bfs(int r, int c) {
       Queue<Position> queue = new LinkedList<>();    // bfs를 수행하기 위함
       List<Position> list = new ArrayList<>();    // 4개 이상 이면 다시 bfs를 수행하지 않기 위해 미리 저장
       queue.offer(new Position(r, c));
       list.add(new Position(r, c));
       visited[r][c] = true;
       char color = map[r][c];
       while (!queue.isEmpty()) {
           Position temp = queue.poll();
   
           for (int i = 0; i < 4; i++) {
               int nr = temp.r + deltas[i][0];
               int nc = temp.c + deltas[i][1];
   
               if (isIn(nr, nc) && map[nr][nc] == color && !visited[nr][nc]) { // 같은 색이고 방문하지 않은 곳
                   list.add(new Position(nr, nc));
                   queue.offer(new Position(nr, nc));
                   visited[nr][nc] = true;
               }
           }
       }
   
       if (list.size() >= 4) {     // List의 크기가 4 이상 == 같은 색 뿌요가 4개이상 붙어있다.
           for (Position temp : list) {
               map[temp.r][temp.c] = '.';
           }
           return true;
       }    
       return false;
   }

2. 중력

   Stack을 쓰면 더 편할 것 같지만 while문으로 구현해보았다.

   static void gravity() {
       for (int j = 0; j < 6; j++) {
           for (int i = 11; i >= 0; i--) {
               if (map[i][j] != '.') {    // 맨 밑부터 탐색하고 뿌요가 나오면 다시 아래로 내려간다
                   int nr = i;
                   while (true) {
                       nr++;
                       if (isIn(nr, j) && map[nr][j] == '.') {
                           map[nr][j] = map[nr - 1][j];
                           map[nr - 1][j] = '.';
                       } else {
                           break;
                       }
                   }
               }
           }
       }
   }

최종 소스코드

import java.io.*;
import java.util.*;

public class BJ_G4_11559_Puyo_Puyo {

    static int answer = 0;
    static char[][] map = new char[12][6];
    static boolean[][] visited = new boolean[12][6];
    static int deltas[][] = { { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };

    static class Position {
        int r, c;

        public Position(int r, int c) {
            super();
            this.r = r;
            this.c = c;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < 12; i++) {
            String str = br.readLine();
            for (int j = 0; j < 6; j++) {
                map[i][j] = str.charAt(j);
            }
        }
        while (true) {
            for (int i = 0; i < 12 ; i++) {
                Arrays.fill(visited[i], false);
            }
            int count = 0;
            for (int i = 0; i < 12; i++) {
                for (int j = 0; j < 6; j++) {
                    if (!visited[i][j] && map[i][j] != '.') {
                        if (bfs(i, j)) {
                            count++;
                        }
                    }
                }
            }

            if (count == 0) {
                break;
            }
            answer++;
            gravity();
        }
        System.out.println(answer);
    }

    static boolean bfs(int r, int c) {
        Queue<Position> queue = new LinkedList<>();
        List<Position> list = new ArrayList<>();
        queue.offer(new Position(r, c));
        list.add(new Position(r, c));
        visited[r][c] = true;
        char color = map[r][c];
        while (!queue.isEmpty()) {
            Position temp = queue.poll();

            for (int i = 0; i < 4; i++) {
                int nr = temp.r + deltas[i][0];
                int nc = temp.c + deltas[i][1];

                if (isIn(nr, nc) && map[nr][nc] == color && !visited[nr][nc]) {
                    list.add(new Position(nr, nc));
                    queue.offer(new Position(nr, nc));
                    visited[nr][nc] = true;
                }
            }
        }

        if (list.size() >= 4) {
            for (Position temp : list) {
                map[temp.r][temp.c] = '.';
            }
            return true;
        }
        return false;
    }

    static void gravity() {
        for (int j = 0; j < 6; j++) {
            for (int i = 11; i >= 0; i--) {
                if (map[i][j] != '.') {
                    int nr = i;
                    while (true) {
                        nr++;
                        if (isIn(nr, j) && map[nr][j] == '.') {
                            map[nr][j] = map[nr - 1][j];
                            map[nr - 1][j] = '.';
                        } else {
                            break;
                        }
                    }
                }
            }
        }
    }

    static boolean isIn(int r, int c) {
        return r >= 0 && r < 12 && c >= 0 && c < 6;
    }
}