[파이썬/프로그래머스]방금그곡

링크> programmers.co.kr/learn/courses/30/lessons/17683

코딩테스트 연습 - [3차] 방금그곡

 

문제>

 

라디오를 자주 듣는 네오는 라디오에서 방금 나왔던 음악이 무슨 음악인지 궁금해질 때가 많다. 그럴 때 네오는 다음 포털의 '방금그곡' 서비스를 이용하곤 한다. 방금그곡에서는 TV, 라디오 등에서 나온 음악에 관해 제목 등의 정보를 제공하는 서비스이다.

네오는 자신이 기억한 멜로디를 가지고 방금그곡을 이용해 음악을 찾는다. 그런데 라디오 방송에서는 한 음악을 반복해서 재생할 때도 있어서 네오가 기억하고 있는 멜로디는 음악 끝부분과 처음 부분이 이어서 재생된 멜로디일 수도 있다. 반대로, 한 음악을 중간에 끊을 경우 원본 음악에는 네오가 기억한 멜로디가 들어있다 해도 그 곡이 네오가 들은 곡이 아닐 수도 있다. 그렇기 때문에 네오는 기억한 멜로디를 재생 시간과 제공된 악보를 직접 보면서 비교하려고 한다. 다음과 같은 가정을 할 때 네오가 찾으려는 음악의 제목을 구하여라.

 

접근>

 

1.  만약 m(melody)이 'CC#'이라면 ['C', 'C#'] 또는 ['C', 'c']과 같이 각 음을 나누어줘야 한다. → tokenizing(string)

2. 시간 계산은 다 분 단위로 바꿔서 하는게 편리하다. → convert(t)

3. 조건이 일치하는 음악이 여러 개일 때에는 라디오에서 재생된 시간이 제일 긴 음악 제목을 반환한다. → (len(v) < len(value)

4. 재생된 시간도 같을 경우 먼저 입력된 음악 제목을 반환한다. → 이건 파이썬이 알아서 해준다

 

주의>

 

문제 설명을 보면

 

• 네오가 기억한 멜로디와 악보에 사용되는 음은 C, C#, D, D#, E, F, F#, G, G#, A, A#, B 12개이다.

 

위와 같이 #이 있는 음은 C, D, F, G, A 의 5개 뿐이지만, 아마 이건 오타다. E#도 있다고 가정하고 풀어야 답으로 인정된다(27번 테스트케이스).

 

코드>

def solution(m, musicinfos):
    def tokenizing(string):
        chars = {'C#': 'c',
                 'D#': 'd',
                 'F#': 'f',
                 'G#': 'g',
                 'A#': 'a',
                 'E#': 'e'}

        melody = []
        for i in range(len(string)-1):
            if string[i+1] == '#':
                melody.append(chars[string[i:i+2]])
            elif string[i] != '#':
                melody.append(string[i])
        if string[i+1] != '#':
            melody.append(string[i+1])
        return ''.join(melody)

    def convert(t):
        hour, minute = t.split(':')
        return (int(hour)*60)+int(minute)

    m = tokenizing(m)
    song = dict()
    for info in musicinfos:
        start, end, title, melody = info.split(',')
        start, end, melody = convert(start), convert(end), tokenizing(melody)
        elapsed, n = end-start, len(melody)
        melody = melody * (elapsed//n) + melody[:elapsed % n]
        song[title] = melody

    k = v = ''
    for key, value in song.items():
        if (m in value) and (len(v) < len(value)):
            k, v = key, value
    return '(None)' if not len(k) else k