라디오를 자주 듣는 네오는 라디오에서 방금 나왔던 음악이 무슨 음악인지 궁금해질 때가 많다. 그럴 때 네오는 다음 포털의 '방금그곡' 서비스를 이용하곤 한다. 방금그곡에서는 TV, 라디오 등에서 나온 음악에 관해 제목 등의 정보를 제공하는 서비스이다.
네오는 자신이 기억한 멜로디를 가지고 방금그곡을 이용해 음악을 찾는다. 그런데 라디오 방송에서는 한 음악을 반복해서 재생할 때도 있어서 네오가 기억하고 있는 멜로디는 음악 끝부분과 처음 부분이 이어서 재생된 멜로디일 수도 있다. 반대로, 한 음악을 중간에 끊을 경우 원본 음악에는 네오가 기억한 멜로디가 들어있다 해도 그 곡이 네오가 들은 곡이 아닐 수도 있다. 그렇기 때문에 네오는 기억한 멜로디를 재생 시간과 제공된 악보를 직접 보면서 비교하려고 한다. 다음과 같은 가정을 할 때 네오가 찾으려는 음악의 제목을 구하여라.
(None)
”을 반환한다.입력으로 네오가 기억한 멜로디를 담은 문자열 m
과 방송된 곡의 정보를 담고 있는 배열 musicinfos
가 주어진다.
m
은 음 1
개 이상 1439
개 이하로 구성되어 있다.musicinfos
는 100
개 이하의 곡 정보를 담고 있는 배열로, 각각의 곡 정보는 음악이 시작한 시각, 끝난 시각, 음악 제목, 악보 정보가 ',
'로 구분된 문자열이다.HH:MM
형식이다.,
' 이외의 출력 가능한 문자로 표현된 길이 1
이상 64
이하의 문자열이다.1
개 이상 1439
개 이하로 구성되어 있다.조건과 일치하는 음악 제목을 출력한다.
m | musicinfos | answer |
---|---|---|
"ABCDEFG" | ["12:00,12:14,HELLO,CDEFGAB", "13:00,13:05,WORLD,ABCDEF"] | "HELLO" |
"CC#BCC#BCC#BCC#B" | ["03:00,03:30,FOO,CC#B", "04:00,04:08,BAR,CC#BCC#BCC#B"] | "FOO" |
"ABC" | ["12:00,12:14,HELLO,C#DEFGAB", "13:00,13:05,WORLD,ABCDEF"] | "WORLD" |
첫 번째 예시에서 HELLO는 길이가 7분이지만 12:00부터 12:14까지 재생되었으므로 실제로 CDEFGABCDEFGAB로 재생되었고, 이 중에 기억한 멜로디인 ABCDEFG가 들어있다.
세 번째 예시에서 HELLO는 C#DEFGABC#DEFGAB로, WORLD는 ABCDE로 재생되었다. HELLO 안에 있는 ABC#은 기억한 멜로디인 ABC와 일치하지 않고, WORLD 안에 있는 ABC가 기억한 멜로디와 일치한다.
def time2min(timestamp: str):
HH, MM = map(int, timestamp.split(":"))
return HH*60+MM
def encode(code: str):
code = code.replace('B#', 'b').replace('C#', '0').replace('D#', '1').replace('F#', '2').replace('G#', '3').replace('A#', '4')
return code
def solution(m, musicinfos):
answers = []
m = encode(m)
for idx, musicinfo in enumerate(musicinfos):
start, end, name, code = musicinfo.split(',')
code = encode(code)
delta = time2min(end) - time2min(start)
share, remainder = delta//len(code), delta%len(code)
total_code = code*share + code[:remainder]
# print(total_code)
if m in total_code:
answers.append((delta, name, -idx))
if answers:
answer = max(answers, key=lambda x: (x[0], x[2]))
return answer[1]
else:
return "(None)"
풀이는 두 가지를 고민하면 된다.
1. 전체 재생된 멜로디를 찾아내기
2. 골라내기(엣지 케이스의 경우 특정 조건일 때 정렬하기)
원래 NLP를 해서 그런가 보자마자 토크나이져가 떠올라버린... 그래서 원래는encode
에서 앞에서부터 읽으면서 #
이 붙는 모든 캐릭터를 다른 숫자로 맵핑하도록 했지만, 생각해 보니 그냥 특수 조건인 (E#
) 숫자들만 한 캐릭터로 표현하면 된다. 문제는 이렇게 했더니 정답이 처음에는 안 나왔다는 것인데,
저기서 B#은 사실 문제 설명에는 없는 것으로, 추가된 test case에 오류가 있어서 따로 집어넣었다. 저게 있어서 이유도 모르고 시간만 보냄..
골라내는 것은 쉽다. 처음에는 마지막 B#
부분이 인덱스 정렬 부분인 줄 알고 추가로 넣어줬는데, max특성상 이미 저걸 해 주고 있어서 사실상 필요 없는 코드 부분.