31562번: 전주 듣고 노래 맞히기

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

 

언어 : Kotlin

 

문제 설명 :

윤수와 정환은 「전주 듣고 노래 맞히기」라는 게임을 할 예정이다. 「전주 듣고 노래 맞히기」는 주어진 노래의 전주를 듣고 먼저 제목을 맞히는 사람이 점수를 얻어 최종적으로 점수가 더 많은 사람이 이기는 게임이다. 절대 음감을 가진 윤수는 노래의 첫 네 음만 듣고도 어떤 노래든 바로 맞힐 수 있다. 따라서, 정환은 윤수를 이기기 위해 첫 세 음만으로 노래를 맞히게 해주는 프로그램을 만들려고 한다. 우선 정환이 알고 있는 노래 제목, 음이름 등을 데이터로 만든 뒤 프로그램을 구현하기 시작했다. 예를 들어, 다음은 TwinkleStar(반짝반짝 작은 별)의 악보 중 일부이다.

위 악보를 박자와 관계없이 음이름으로 표현하면 CCGGAAG가 된다.

윤수를 이기기 위해서는 이 프로그램이 첫 세 음인 CCG만으로 노래 제목인 TwinkleStar를 출력할 수 있어야 한다. 또한, 세상의 모든 노래를 아는 윤수와 다르게 정환은 음을 아는 노래가 
$N$개뿐이다. 그래서 프로그램에 
$N$개의 노래의 정보를 저장해 놓을 것이다. 만약 저장된 노래 중 입력한 첫 세 음으로 시작하는 노래가 여러 개 있어 무슨 노래인지 정확히 알 수 없는 경우 ?를 출력하고, 입력한 첫 세 음에 맞는 저장된 노래가 없을 경우 !를 출력한다.

정환을 도와서 첫 세 음만으로 본인이 음을 아는 노래를 맞히는 프로그램을 완성하자. 이 프로그램은 대문자와 소문자를 구분한다.

 

입력 :

첫 번째 줄에 정환이 음을 아는 노래의 개수 N, 정환이 맞히기를 시도할 노래의 개수 M이 공백으로 구분되어 주어진다.

두 번째 줄부터 N개의 줄에 걸쳐 노래 제목의 길이 T, 영어 대소문자로 이루어진 문자열 노래 제목 S, 해당 노래에서 처음 등장하는 일곱 개의 음이름 a_1, a_2, a_3, a_4, a_5, a_6, a_7이 공백으로 구분되어 주어진다.

 N+2번째 줄부터 M개의 줄에 걸쳐 정환이 맞히기를 시도할 노래의 첫 세 음의 음이름 b_1, b_2, b_3가 공백으로 구분되어 주어진다.

주어지는 음이름은 각각 C, D, E, F, G, A, B 중 하나이다. 같은 제목이 두 번 이상 주어지지 않는다.

 

출력 :

정환이 맞히기를 시도할 각 노래에 대하여 프로그램에 저장된 노래와 첫 세 음이 동일한 노래가 하나만 있다면 해당 노래의 제목을, 두 개 이상이면 ?을, 없다면 !을 한 줄에 하나씩 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 1024MB
• 1 ≤ N ≤ 1,000
• 1 ≤ M ≤ 1,000
• 1 ≤ T ≤ 30
• 입력으로 주어지는 모든 수는 정수이다.

 

입출력 예 :

입력	출력
4 4 11 TwinkleStar C C G G A A G 8 Marigold E D E F E E D 23 DoYouWannaBuildASnowMan C C C G C E D 12 Cprogramming C C C C C C C E D E C G G C C C C C G	Marigold ! ? TwinkleStar

 

풀이 : 

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

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

    val (n, m) = readLine()
        .split(" ")
        .map { it.toInt() }

    val musicList = MutableList<Music>(n) {
        val cur = readLine().split(" ")
        val notes = cur.subList(cur.size - 7, cur.size).joinToString(" ")
        Music(cur[0].toInt(), cur[1], notes)
    }

    repeat(m) {
        val cur = readLine()
        val filtered = musicList.filter { it.notes.startsWith(cur) }

        if (filtered.size == 1) {
            bw.write(filtered[0].titleName)
        } else if (filtered.isEmpty()) {
            bw.write("!")
        } else {
            bw.write("?")
        }
        bw.newLine()
    }

    bw.flush()
    bw.close()
}

data class Music(
    val titleLength: Int,
    val titleName: String,
    val notes: String
)