[프로그래머스 level2 방금그곡] '차근 차근' 조건을 분석하자!

이 문제는 풀고나서 기분이 굉장히 좋아지는 문제였다.

(조건이 많아서 까다로워보였던 문제였기 때문이다..)

오늘부터는 지금 열심히 인턴으로 일하구 있는 친구들과 

금요일, 토요일 밤마다 문제를 시간 재고 같이 풀고 코드리뷰를 하기로 해서 한 문제만 풀고자 한다!

 

해결해야하는 문제는 다음과 같았다.

programmers.co.kr/learn/courses/30/lessons/17683

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

 

문제 설명↓

 

방금그곡

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

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

• 방금그곡 서비스에서는 음악 제목, 재생이 시작되고 끝난 시각, 악보를 제공한다.
• 네오가 기억한 멜로디와 악보에 사용되는 음은 C, C#, D, D#, E, F, F#, G, G#, A, A#, B 12개이다.
• 각 음은 1분에 1개씩 재생된다. 음악은 반드시 처음부터 재생되며 음악 길이보다 재생된 시간이 길 때는 음악이 끊김 없이 처음부터 반복해서 재생된다. 음악 길이보다 재생된 시간이 짧을 때는 처음부터 재생 시간만큼만 재생된다.
• 음악이 00:00를 넘겨서까지 재생되는 일은 없다.
• 조건이 일치하는 음악이 여러 개일 때에는 라디오에서 재생된 시간이 제일 긴 음악 제목을 반환한다. 재생된 시간도 같을 경우 먼저 입력된 음악 제목을 반환한다.
• 조건이 일치하는 음악이 없을 때에는 “(None)”을 반환한다.

 

입력 형식

입력으로 네오가 기억한 멜로디를 담은 문자열 m과 방송된 곡의 정보를 담고 있는 배열 musicinfos가 주어진다.

• m은 음 1개 이상 1439개 이하로 구성되어 있다.
• musicinfos는 100개 이하의 곡 정보를 담고 있는 배열로, 각각의 곡 정보는 음악이 시작한 시각, 끝난 시각, 음악 제목, 악보 정보가 ','로 구분된 문자열이다.
  • 음악의 시작 시각과 끝난 시각은 24시간 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가 기억한 멜로디와 일치한다.

 

 

무작정 다가가면 안될 것 같은 길이의 문제였다.

 

이 문제에서 조심해야 할 점은 4가지가 있었다.

1. 악보에 사용되는 음이 C, C#, D, D#, E, F, F#, G, G#, A, A#, B로 글자 길이가 다르다는 점

2. 조건에 일치하는 음악이 한 개만 있는 것이 아니라 여러 개가 나올 수 있다는 점

-> 여러 개인 경우, 재생된 시간이 가장 긴 음악, 먼저 입력된 음악 제목 반환

3. 일치하는 음악이 없다면 'None' 문자열이 아닌 '(None)' 문자열을 반환한다는 점

4. 음악 재생 시간 계산 시 00:00 는 고려하지 않아도 된다는 점.

 

 

 

우선, 음악 제목에 따른 재생된 음들을 저장하는 딕셔너리인 music_time_dic을 초기화하였고, 음악이 언제 들어왔는지를 표시하기 위한 music_come_dic 을 초기화하였고, 음악이 재생된 총 재생시간을 저장하는 music_total_dic을 초기화하였다.

 

musicinfos를 하나씩 보며 music_time_dic과 music_come_dic, music_total_dic을 채워나갔다.

musicinfos는 음악 제목, 재생이 시작되고 끝난 시각, 악보를 ','로 구분하였으므로 split 함수를 통해 , 단위로 쪼개어 만든 배열을 temp에 저장하였다.

 

시간은 :을 기준으로 쪼갰고, 00:00 를 넘어가지 않고, 24시간 단위로 시작시간, 끝나는 시간이 들어오므로 재생시간은 다음과 같이 계산하였다.

time1 = temp[0].split(':')
time2 = temp[1].split(':')
total = (int(time2[0])*60 + int(time2[1])) - (int(time1[0])*60 + int(time1[1]))

 

 

 

 

for문을 통해 문자열을 하나씩 검사하고 싶었으나 '#'을 처리할 방법을 생각하지 못하였으므로 while문으로 처리하였다.

재생 시간동안 반복문이 돌아야하므로 count < total 인 경우 반복문이 실행되도록 하였다.

(나중에 다른 사람들의 풀이를 보니 C#은 사전에 소문자c와 같은 것으로 변형하여 처리하였다. 나는 문제 풀 당시에 이를 생각해내지 못하였기 때문에..while문을 사용하였다.)

 

 

j가 악보의 마지막 index라면 j+1을 검사할 때 out of index 이므로, j가 마지막 인덱스가 아닌 경우로 조건을 걸어주었다.

마지막 인덱스가 아닌 경우, 뒤의 문자가 '#'이면 music_time_dic에 시간의 흐름에 따른 해당 곡에 대한 악보를 저장하므로 아이디어를 바탕으로 아래와 같이 하였다.

이 때 딕셔너리의 값을 문자열이 아닌 배열로 한 이유는 C와 C#을 같은 문자열 길이로 인식해주기 위함이였다.

while count<total :
            if j<len(temp[3])-1 :
                if temp[3][j+1] == '#':
                    if temp[2] in music_time_dic :
                        music_time_dic[temp[2]].append(temp[3][j] + '#')
                    else :
                        music_time_dic[temp[2]] = [temp[3][j] + '#']
                    j+=1
                else :
                    if temp[2] in music_time_dic :
                        music_time_dic[temp[2]].append(temp[3][j])
                    else :
                        music_time_dic[temp[2]] = [temp[3][j]]

            else :
                if temp[2] in music_time_dic :
                    music_time_dic[temp[2]].append(temp[3][j])
                else :
                    music_time_dic[temp[2]] = [temp[3][j]]

            count += 1
            j += 1
            if j == len(temp[3]) :
                j=0

 

 

 

 

 

 

반복문을 순회한 후, 음악이 언제들어왔는지에 대한 정보인 music_come_dic와 재생 시간을 담는 music_total_dic을 담아주었다.

 

 

모두 딕셔너리에 저장한 다음에 m의 길이를 구하였고, music_time_dic을 정렬하였다.

music_dic_items()는 music_time_dic 딕셔너리를 (key,value)로 갖는 튜플 쌍이다.

이는 music_total_dic[key] 재생 길이를 역순으로 정렬하고 같은 재생 길이를 갖는 경우, music_come_dic[key] 들어온 순서로 정렬하였다.

여기서 재생시간은 긴 순서고, 음악이 입력된 시간은 먼저 들어온 순서이므로, reverse=True를 사용하기 위해,

위에서 music_come_dic[key]는 'music_come_dic[temp[2]] = len(musicinfos)-i' 로 저장하였다.

music_time_dic = sorted(music_time_dic.items(),key = lambda x : (music_total_dic[x[0]],music_come_dic[x[0]]),reverse=True)

 

 

m과 일치하는 것이 존재하는지 music_time_dic 튜플을 순회하며 같은 것이 있는지 찾았고, 없다면 (None)을 반환하도록 하였다.

 

이러한 아이디어로 푼 최종 코드는 다음과 같다.

def solution(m, musicinfos):
    answer = ''
    music_time_dic = {}
    music_come_dic = {}
    music_total_dic = {}
    for i in range(len(musicinfos)):
        temp = musicinfos[i].split(',')
        time1 = temp[0].split(':')
        time2 = temp[1].split(':')
        total = (int(time2[0])*60 + int(time2[1])) - (int(time1[0])*60 + int(time1[1]))
        j=0
        count = 0
        while count<total :
            if j<len(temp[3])-1 :
                if temp[3][j+1] == '#':
                    if temp[2] in music_time_dic :
                        music_time_dic[temp[2]].append(temp[3][j] + '#')
                    else :
                        music_time_dic[temp[2]] = [temp[3][j] + '#']
                    j+=1
                else :
                    if temp[2] in music_time_dic :
                        music_time_dic[temp[2]].append(temp[3][j])
                    else :
                        music_time_dic[temp[2]] = [temp[3][j]]

            else :
                if temp[2] in music_time_dic :
                    music_time_dic[temp[2]].append(temp[3][j])
                else :
                    music_time_dic[temp[2]] = [temp[3][j]]

            count += 1
            j += 1
            if j == len(temp[3]) :
                j=0
        music_come_dic[temp[2]] = len(musicinfos)-i
        music_total_dic[temp[2]] = total

    m_len = len(m) - m.count('#')
    flag=0
    music_time_dic = sorted(music_time_dic.items(),key = lambda x : (music_total_dic[x[0]],music_come_dic[x[0]]),reverse=True)
    for i in music_time_dic:
        for j in range(len(i[1])) :
            if m == ''.join(i[1][j:j+ m_len]):
                flag=1
                break
        if flag ==1 :
            answer = i[0]
            break
    else :
        answer = '(None)'

    return answer

 

최종 코드를 보니 굉장히 길었다.

 

다른 사람들의 풀이를 보니 C#이 들어가면 c로 수정하고 D#은 d로 문자열을 replace하는 방식으로 글자수를 맞추어서 처리한 경우가 가장 많았다. 

또한 music을 class로 만들어 각 음악이 ','로 split한 정보를 담고 있도록 하여 처리한 경우도 있었다. 

이 두 경우 모두 좋은 방안이라고 생각이 들었다 ❗️

 

앞으로 class를 활용해야 한다면 사용해봐야겠다..ㅎㅎ

좀 있다가 시작될 친구들과의 코딩을 위해 머리를 잠깐 식혀야겠다.⭐️