21736번: 헌내기는 친구가 필요

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

21736번: 헌내기는 친구가 필요해

언어 : Kotlin

 

문제 설명 :

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개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.
• 첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.

 

제한 사항 :

• 시간 제한 : 1초 (하단 참고)
• 메모리 제한 : 1024MB

• Java 8: 2 초
• Python 3: 2 초
• PyPy3: 2 초
• Java 8 (OpenJDK): 2 초
• Java 11: 2 초
• Python 2: 2 초
• PyPy2: 2 초
• Kotlin (JVM): 2 초
• Java 15: 2 초

 

입출력 예 :

입력	출력
3 5 OOOPO OIOOX OOOXP	1
3 3 IOX OXP XPP	TT

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

var result = 0

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val (n, m) = br.readLine().split(" ").map { it.toInt() }
    val map = Array(n) { Array(m) { "" } }
    val player = intArrayOf(0, 0)
    repeat(n) { i ->
        br.readLine().also {
            for (j in it.indices) {
                map[i][j] = it[j].toString()
                if (map[i][j] == "I") { player[0] = i; player[1] = j; }
            }
        }
    }

    val dx = intArrayOf(-1, 1, 0, 0)
    val dy = intArrayOf(0, 0, -1, 1)
    val visited = Array(n) { BooleanArray(m) { false } }
    dfs(n, m, player[0], player[1], dx, dy, visited, map)

    if (result == 0) bw.write("TT") else bw.write("$result")
    bw.flush()
    bw.close()
}

fun dfs(n: Int, m: Int, x: Int, y: Int, dx: IntArray, dy: IntArray, visited: Array<BooleanArray>, map: Array<Array<String>>) {
    visited[x][y] = true

    if (map[x][y] == "P") result++

    for (i in 0 until 4) {
        val nx = x + dx[i]
        val ny = y + dy[i]

        if (nx >= 0 && ny >= 0 && nx < n && ny < m && !visited[nx][ny]) {
            if (map[nx][ny] != "X") { dfs(n, m, nx, ny, dx, dy, visited, map) }
        }
    }
}

 

이전에 풀었던 토마토와 유사한 문제

https://small-stepping.tistory.com/533

7576번: 토마토