[백준] 11559번 Puyo Puyo

백준 11559번 - Puyo Puyo

문제

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

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

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

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

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

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

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

접근 방법

  1. 지울 수 있는 영역 파악
    유의사항
    • 지울 수 있는 영역을 파악하기 위해 완전탐색 + bfs를 이용하였다.
  2. 영억 지우기
  3. 비워진 영역 채우기

코드 개선점

  1. visited 배열을 매번 생성하지 않고 연쇄횟수를 값으로 받아 재사용하는 방법 존재
  2. 지울 수 있는 영역을 파악하고 조건 만족 시 remove를 바로 함으로써 메모리 사용량 감소

코드

import java.util.Queue
import java.util.LinkedList

class IO11559 {
    private val br = System.`in`.bufferedReader()
    private val bw = System.out.bufferedWriter()

    fun getRow() = br.readLine().toList().map { it.toString() }.toMutableList()
    fun close() = bw.close()
    fun write(message: String) = bw.write(message)
}

fun main() {
    var answer = 0
    val io = IO11559()
    val graph = Array(12){ mutableListOf<String>()}
    repeat(12) {
        graph[it] = io.getRow()
    }

    val colGraph = Array(6) { mutableListOf<String>() }
    repeat(6) {
        val tmpList = mutableListOf<String>()
        for (idx in 11 downTo 0) {
            tmpList.add(graph[idx][it])
        }
        colGraph[it] = tmpList
    }

    val dxy = listOf( -1 to 0, 1 to 0, 0 to -1, 0 to 1)

    data class Point(val col: Int, val row: Int)

    fun findErasableArea(): MutableList<MutableSet<Point>> {
        val visited = Array(6) { Array(12) { false } }
        val eraseAreaList = mutableListOf<MutableSet<Point>>()
        for (col in (0..5)) {
            for (row in (0..11)) {
                if (visited[col][row]) continue
                if (colGraph[col][row] ==".") {
                    visited[col][row] = true
                    continue
                }
                val q: Queue<Point> = LinkedList()
                val tmpEraser = mutableSetOf<Point>()

                q.add(Point(col, row))
                tmpEraser.add(Point(col, row))

                while (q.isNotEmpty()) {
                    val (curCol, curRow) = q.poll()
                    if (visited[curCol][curRow]) continue
                    visited[curCol][curRow] = true
                    for (idx in 0..3) {
                        val (cy, cx) = dxy[idx]
                        val nx = cx + curRow
                        val ny = cy + curCol
                        if (nx < 0 || nx > 11 || ny < 0 || ny > 5) continue
                        if (colGraph[ny][nx] == colGraph[col][row] && !visited[ny][nx]) {
                            Point(ny,nx).also {
                                q.add(it)
                                tmpEraser.add(it)
                            }
                        }
                    }
                }
                if (tmpEraser.size > 3) {
                    eraseAreaList.add(tmpEraser)
                }
            }
        }
        return eraseAreaList
    }

    fun remove(area: MutableList<MutableSet<Point>>) {
        area.forEach { each ->
            each.forEach { it ->
                val (col, row) = it
                colGraph[col][row] = "."
            }
        }
    }

    fun fillEmpty() {
        for (col in 0..5) {
            val tmpCol = colGraph[col].filter { it != "." }.toMutableList()
            for (r in 1..(12-tmpCol.size)) {
                tmpCol.add(".")
            }
            colGraph[col] = tmpCol
        }
    }

    while (true) {
        val erasableAreaList = findErasableArea()
        if (erasableAreaList.isEmpty()) {
            break
        }
        remove(erasableAreaList)
        fillEmpty()
        answer++
    }

    io.write(answer.toString())
    io.close()

}

좋은 웹페이지 즐겨찾기