[Kotlin] BOJ 21736. 헌내기는 친구가 필요해

출처

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

 

 

 

문제

2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다. 

도연이가 다니는 대학의 캠퍼스는 N×M 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (x, y)에 있다면 이동할 수 있는 곳은 (x+1, y), (x, y+1), (x−1, y), (x, y−1)이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.

불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.

 

 

입력

첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 N (1≤N≤600), M (1≤M≤600)이 주어진다.

둘째 줄부터 N$N$개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.

 

 

출력

첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.

 

 

 

풀이

package `20001~25000`

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.LinkedList
import java.util.Queue
import java.util.StringTokenizer

private var N = 600
private var M = 600
private var campus = Array(N){ MutableList<String>(M) { "" } }
private var visited = Array(N){ MutableList<Boolean>(M) { false } }
private var cnt = 0
private val queue : Queue<Pair<Int, Int>> = LinkedList()
private val dx = listOf(-1, 1, 0, 0)
private val dy = listOf(0, 0, -1, 1)
fun bfs() {
    while (queue.isNotEmpty()) {
        val current = queue.poll()
        val x = current.first
        val y = current.second
        for (i in 0 until 4) {
            val nx = x + dx[i]
            val ny = y + dy[i]
            if (nx in 0 until N && ny in 0 until M  && campus[nx][ny] != "X" && !visited[nx][ny]) {
                if (campus[nx][ny] == "P") {
                    cnt += 1
                }
                visited[nx][ny] = true
                queue.offer(Pair(nx, ny))
            }
        }
    }
}

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val st = StringTokenizer(br.readLine())
    N = st.nextToken().toInt()
    M = st.nextToken().toInt()
    repeat(N) { n ->
        val line = br.readLine().toCharArray().map { it.toString() }
        campus[n] = line.toMutableList()
        repeat(M) { m ->
            if (campus[n][m] == "I") {
                queue.offer(Pair(n, m))
            }
        }
    }
    bfs()
    if (cnt == 0) {
        println("TT")
        return
    }
    println("$cnt")
}