99클럽 코테 스터디 9일차 TIL + 딕셔너리, join(), 리스트

1. 문제 설명

[문제 내용]

윤수와 정환은 「전주 듣고 노래 맞히기」라는 게임을 할 예정이다. 「전주 듣고 노래 맞히기」는 주어진 노래의 전주를 듣고 먼저 제목을 맞히는 사람이 점수를 얻어 최종적으로 점수가 더 많은 사람이 이기는 게임이다. 절대 음감을 가진 윤수는 노래의 첫 네 음만 듣고도 어떤 노래든 바로 맞힐 수 있다. 따라서, 정환은 윤수를 이기기 위해 첫 세 음만으로 노래를 맞히게 해주는 프로그램을 만들려고 한다. 우선 정환이 알고 있는 노래 제목, 음이름 등을 데이터로 만든 뒤 프로그램을 구현하기 시작했다. 예를 들어, 다음은 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 \le N \le 1\,000$ 

• $1 \le M \le 1\,000$ 

• $1 \le T \le 30$ 

• 입력으로 주어지는 모든 수는 정수이다.

[입출력 예]

2. 풀이

import sys
input = sys.stdin.readline

N, M = map(int, input().split())

dic_song = {}  
for _ in range(n):
    song = input().split()
    sound = ''.join(song[2:5])

    if sound in dic_song:   
        dic_song[sound] = '?'
    else:
        dic_song[sound] = song[1]

for _ in range(m):
    sound = ''.join(input().split())

    if sound in dic_song:
        print(dic_song[sound])
    else:
        print('!')

3. 회고

3-1. 문제 해결 과정

역시 입력까지는 문제 없었지만, 음 이름을 어떻게 처리해야 하나 고민이 많았다.

공백으로 구분하는데 뒤는 어떻게 하지? 하고 있었는데 생각해보니 우리는 앞에 3개 자리만 구분하기로 했었다. 그래서 sound = ''.join(song[2:5])을 이용해서 뒤에 있는 음 이름 3개는 붙여서 저장하게 되었다.

그리고 해당 음계가 없으면 dic_song에 추가한다. dic_song[sound] = song[1]을 이용해서 딕셔너리에 저장하게 되니까, 딕셔너리에는 "CCG" : "TwinkleStar" 형태로 저장된다.

그리고 중복으로 저장되었을 때는 똑같이 저장하는게 아닌 dic_song[sound] = '?'을 이용해서 저장하므로 중복 처리도 쉽게 할 수 있다.

3-2. 새롭게 배운 내용

• ''.join(input(song[2:5]) : 이를 이용하면 공백을 기준으로 하여 해당 인덱스 만큼의 내용을 붙여서 저장할 수 있다.