Q.11559 자바 : Puyo Puyo(G4)

문제 : https://www.acmicpc.net/problem/11559

 

일하면서 풀어서 2일정도 걸려서 풀게 된 문제이다.

이 문제를 풀고 있을 때 코딩테스트를 모르는 친구가 코딩테스트가 뭐냐? 라고 물어본적이 있는데

이 문제 설명을 읽게 하니까 이해를 했다고 하면서 왜 이런걸 푸냐고 하긴 했다.

 

개인적으로 푸는 과정 자체는 상당히 고통스럽기도 하고 한 2~3일정도(일하면서) 박치기를 하다 보면

찾아보고 싶은 욕구를 참기가 꽤나 힘들다.

하지만 풀었을 때의 그 성취감이 상당하기 때문에 그리고 코딩테스트를 보는 기업이 많으므로 준비하고 있다.

---

문제

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

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

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

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

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

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

 

남규는 최근 뿌요뿌요 게임에 푹 빠졌다. 이 게임은 1:1로 붙는 대전게임이라 잘 쌓는 것도 중요하지만,

상대방이 터뜨린다면 연쇄가 몇 번이 될지 바로 파악할 수 있는 능력도 필요하다.

하지만 아직 실력이 부족하여 남규는 자기 필드에만 신경 쓰기 바쁘다.

상대방의 필드가 주어졌을 때, 연쇄가 몇 번 연속으로 일어날지 계산하여 남규를 도와주자!

 

입력

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

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

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

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

 

출력

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

처음 문제를 봤을 때 떠올린 내용을 작성해보자면

1. 색깔별로 나눠서 계산

2. 색깔별로 4개가 이웃한 색인지 판별하는 알고리즘은 BFS를 사용

3. 이 때 순회중인 내용에 대해 boolean 배열로 순회중이라고 표시

4. 4개 이상의 이웃한 색일 경우 Pang() 하는 로직을, 아닐 경우 되돌리는 로직 작성(Pang(), RollbackColor() )

5. 문제에서 터질 수 있는 뿌요가 여러 그룹이 있다면 동시에 터져야 하고

여러 그룹이 터지더라도 한번의 연쇄가 추가된다고 했으므로 모든 순회가 끝난 뒤에 중력을 적용(afterPang() )

 

이정도로  생각해 작성했다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

class Main {
    static BufferedReader br;
    static StringBuffer sb;
    static char[][] games;
    static boolean[][] curGameColor;

    static int[] dx = {1, 0, -1, 0};
    static int[] dy = {0, 1, 0, -1};
    static boolean isPang = true;
    static boolean curColorPang;
    static int pangCount = 0;

    static int curPangStack = 1;

    enum Color {R, G, B, P, Y}

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

        games = new char[12][6];
        curGameColor = new boolean[12][6];
        for (int i=0; i<12; i++){
            games[i] = br.readLine().toCharArray();
        }

        while (isPang) {
            isPang = false;
            for(Color color : Color.values()) {
                initCurColorGame(color);
                curColor = color;
                curColorPang = false;
                for (int y=0; y<12; y++) {
                    for (int x = 0; x < 6; x++) {
                        if (curGameColor[y][x]) {
                            curPangStack = 1;
                            BFS(y, x);
                            if (curColorPang) {
                                Pang();
                                isPang = true;
                                curColorPang = false;
                                curPangStack = 1;
                            } else {
                                RollbackColor(curColor);
                            }
                        }
                    }
                }
            }

            if (isPang) {
                pangCount++;
                for (int y = 11; y > 0; y--) {
                    for (int x = 0; x < 6; x++) {
                        AfterPang(y, x);
                    }
                }
            }
        }
        System.out.println(pangCount);
    }
    static void initCurColorGame(Color color){
        char curColor = color.toString().charAt(0);
        for (int y=0; y<12; y++) {
            for (int x=0; x<6; x++) {
                if(games[y][x] == curColor) {
                    curGameColor[y][x] = true;
                }else {
                    curGameColor[y][x] = false;
                }
            }
        }
    }

    static void BFS(int y, int x) {

        curGameColor[y][x] = !curGameColor[y][x];
        games[y][x] = 'X';

        for (int i=0; i<4; i++) {
            int prevY = y + dy[i];
            int prevX = x + dx[i];
            if (prevY >= 0 && prevX >= 0 && prevY < 12 && prevX < 6) {
                if (curGameColor[prevY][prevX]) {
                    curPangStack++;
                    if (curPangStack >= 4) {
                        curColorPang = true;
                    }
                    BFS(prevY, prevX);
                }
            }
        }
    }


    static void RollbackColor(Color color) {
        for (int y=0; y<12; y++) {
            for (int x=0; x<6; x++) {
                if (games[y][x] == 'X') {
                    games[y][x] = color.toString().charAt(0);
                }
            }
        }
    }
    static void Pang() {
        for (int y=0; y<12; y++) {
            for (int x=0; x<6; x++) {
                if (games[y][x] == 'X') {
                    games[y][x] = '.';
                }
            }
        }
    }

    static void AfterPang(int y, int x) {
        if (games[y][x] == '.') {
            for(int i = y-1; i >= 0; i--) {
                if (games[i][x] != '.') {
                    games[y][x] = games[i][x];
                    games[i][x] = '.';
                    return;
                }
            }
        }
    }

}

사실 블로그를 작성하면서 코드를 보니 좀 더 줄일 수 있는 부분이 보인다.

다음에 코드를 줄여서 작성해서 다시 제출한 뒤 글을 추가해서 작성해야겠다.