[백준] 메시지 (1384)(kotlin)

문제 설명

백준 1384번 문제 링크

입력 및 출력

» 입력

• 입력은 번호를 가진 그룹들로 이루어져 있습니다. 그룹의 번호는 1부터 시작하며 1씩 증가합니다. 각 그룹의 입력은 참여한 아이들의 수 n(5 ≤ n ≤ 20)이 한 줄에 주어지면서 시작됩니다. 다음 n 줄에 걸쳐 위 활동을 끝마친 종이 n장이 입력됩니다. n 줄의 순서는 아이들이 앉아있고 종이를 넘기던 순서와 같습니다. 예를 들어 첫 번째 줄에 해당하는 아이는 두 번째 줄에 해당하는 아이에게 종이를 넘겼고, 마지막 줄에 해당하는 아이는 첫 번째 줄에 해당하는 아이에게 종이를 넘겼습니다. 각 줄은 종이 맨 위의 이름으로 시작합니다. 다음으로 공백을 사이에 두고 종이에 적힌 메시지가 종이 위에서부터 아래로 순서대로 쓰입니다. 문제를 간단하게 하기 위해 좋은 메시지는 ‘P’로, 나쁜 메시지는 ‘N’으로 표기합니다. 이름은 60글자 이하의 알파벳 소문자 또는 대문자로 이루어져 있습니다. 마지막 줄에 ‘0’이 입력되면서 입력이 종료됩니다. ‘0’은 처리하지 않습니다.

» 출력

• 그룹 번호를 “Group 1”과 같이 출력함으로써 출력을 시작합니다. 그 다음 줄부터 누가(A) 누구(B)에게 나쁜 말을 했는지 “A was nasty about B”로 한 줄씩 출력합니다. 나쁜 말이 여러 개라면, 입력받은 순서대로─첫 번째 종이부터, 왼쪽에서 오른쪽으로─출력합니다. 아무도 나쁜 말을 하지 않았다면 “Nobody was nasty”를 출력합니다. 각 그룹을 빈 줄로 구분합니다.

예제 입출력(테스트케이스)

입력	출력
5 Ann P N P P Bob P P P P Clive P P P P Debby P N P P Eunice P P P P 6 Zheng P P P P P Yeng P P P P P Xiao P P P P P Will P P P P P Veronica P P P P P Utah P P P P P 0	Group 1 Debby was nasty about Ann Bob was nasty about Debby Group 2 Nobody was nasty

문제 풀이1

fun main(args: Array<String>) = with(System.`in`.bufferedReader()) {
    var group = 1

    while (true) {
        val n = readLine().toInt()
        if (n == 0) break

        println("Group ${group++}")
        val list = Array(n) { readLine().split(" ") }
        var check = false
        list.forEachIndexed  { idx, arr ->
            arr.forEachIndexed { jdx, ch ->
                if (jdx != 0 && ch == "N") {
                    check = true
                    val obj = idx // obj : 2, jdx : 5, n : 6
                    val subj = if (0 > obj - jdx) n + (obj - jdx) else obj - jdx

                    println("${list[subj][0]} was nasty about ${list[obj][0]}")
                }
            }
        }

        if (!check) println("Nobody was nasty")
        println()
    }
}

// 6
// Zheng P P P P P
// Yeng P P P P N
// Xiao P P P P P
// Will N P P P P
// Veronica P P P P P
// Utah P P P P P
// -> Xiao was nasty about Yeng
// -> Xiao was nasty about Will