백준 11559 Puyo Puyo Kotlin (시뮬레이션)

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

11559번: Puyo Puyo

문제

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

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

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

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

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

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

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

입력

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

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

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

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

출력

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

알고리즘 분류

• 구현
• 그래프 이론
• 그래프 탐색
• 너비 우선 탐색
• 시뮬레이션

풀이

dfs 혹은 bfs + 구현으로 풀 수 있다.

지극히 개인적으로 이런 코드가 길어지는 문제는 난이도 하나씩 더 올려줘도 되지 않을까?

본인은 dfs로 풀었고, 오래 걸렸다.

 

풀이를 정리하면 다음과 같다.

1. 그래프 전체를 탐색하면서, '.'이 아닌 칸을 만나면 4개 이상인지 탐색을 시작한다.(bfs혹은 dfs)

2. 1.에서 방문한 칸은 idx로 넘버링 해주면서 구별해주고, 만약 4개 이상이 붙어있다면 idx와 같은 칸들을 모두 '.'으로
   바꾼다.(Bomb)

3. 1,2의 과정으로 한 턴에 터질 뿌요들을 모두 터쳤다. 이제 뿌요들을 아래로 내려주는 것을 구현하자.

4. 1,2 과정에서 4개 이상인 뿌요가 없었다면 전체 반복문을 종료한다.

5. 아래 코드에선 마지막에 턴이 종료되어도 ans를 증가시켜주고 반복문을 종료했기 때문에 ans-1을 출력한다.

 

그래프의 크기가 작아서 그래프 전체 탐색을 부담 없이 할 수 있었다.

코드

val dirXY = arrayOf(arrayOf(0,1),arrayOf(0,-1),arrayOf(1,0),arrayOf(-1,0))

val br = System.`in`.bufferedReader()
var cnt=0
fun dfs(r : Int, c : Int, color : Char, idx : Int,graph : Array<CharArray>, visited : Array<IntArray>) : Boolean{

    var isBomb = false
    visited[r][c]=idx
    if(cnt>=4) isBomb=true
    for(i in 0 until 4){
        val nr = r+dirXY[i][0]
        val nc = c+dirXY[i][1]
        if(nr !in 0 until 12 || nc !in 0 until 6) continue
        if(graph[nr][nc]!=color) continue
        if(visited[nr][nc]==idx)continue
        visited[nr][nc]=idx
        cnt++
        isBomb= dfs(nr,nc,color,idx,graph,visited)
    }

    return isBomb
}

fun bomb(idx : Int, graph : Array<CharArray>, visited: Array<IntArray>){
    for(r in 0 until 12){
        for(c in 0 until 6){
            if(visited[r][c]==idx)
                graph[r][c]='.'
        }
    }
}

fun main() = with(System.out.bufferedWriter()){
    val graph = Array(12){br.readLine().toCharArray()}
    var finish =false
    var ans=0
    while(!finish){
        //RGBPY
        finish =true
        var idx=1
        val visited =Array(12){IntArray(6)}
        
        //bomb
        for(r in 0 until 12){
            for(c in 0 until 6){
                if(graph[r][c]!='.' && visited[r][c]==0){
                    cnt=1
                    if(dfs(r,c,graph[r][c],idx,graph,visited)){
                        finish=false
                        bomb(idx,graph,visited)
                    }
                    idx++
                }
            }
        }
        //내리기
        for(c in 0 until 6){
            for(r in 11 downTo 0){
                if(graph[r][c]=='.'){
                    var idx =r
                    while(idx>=0){
                        if(graph[idx][c]!='.')break
                        idx--
                    }
                    if(idx<0){
                        break
                    }
                    else{
                        graph[r][c]=graph[idx][c].also{graph[idx][c]='.'}
                    }
                }
            }
        }
        ans++
    }
    write("${ans-1}")
    close()
}