알고리즘 & 코딩테스트

[프로그래머스] 2018 KAKAO BLIND RECRUITMENT 방금그곡

Haksae 2022. 3. 6. 17:19

[방금그곡]

라디오를 자주 듣는 네오는 라디오에서 방금 나왔던 음악이 무슨 음악인지 궁금해질 때가 많다. 그럴 때 네오는 다음 포털의 '방금그곡' 서비스를 이용하곤 한다. 방금그곡에서는 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개 이하로 구성되어 있다.

출력 형식

조건과 일치하는 음악 제목을 출력한다.

입출력 예시

mmusicinfosanswer
"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가 기억한 멜로디와 일치한다.

 

👉🏻  답안

import math

def solution(m, musicinfos):
  answer = []
  
  def time(start, end):          # 재생된 시간 구하는 함수
    if int(end[0]) - int(start[0]) == 0: # 라디오에서 재생될 때 시간이 넘어가지 않은 경우
      playtime = int(end[1]) - int(start[1]) # 그냥 분 단위만 계산
      return playtime

    else :  # 라디오에서 재생될 때 시간이 넘어간 경우 
      hour = int(end[0]) - int(start[0]) # 시간 계산
      playtime = hour*60 + int(end[1]) - int(start[1]) # 시간과 분 단위 계산
      return playtime
  
  for i in range(len(musicinfos)): # musicinfos만큼 for 문
  
    song = musicinfos[i].split(',')  # one of musicinfos
    start = song[0].split(':')  # 노래 시작한 시간
    end = song[1].split(':')  # 노래 끝난 시간
    temp_note = song[3]  # 악보에 사용된 음
    title = song[2]      # 노래 제목

	#  #=>소문자 치환
    note = temp_note.replace("C#", "c").replace("D#","d").replace("F#","f").replace("G#", "g").replace("A#","a")
    search = m.replace("C#", "c").replace("D#","d").replace("F#","f").replace("G#", "g").replace("A#","a")

    realtime = time(start,end) # time 함수 이용하여 실제 시간 구함
    fullnote = note * (math.ceil(realtime / len(note))) # 재생된 시간 길이를 악보 길이로 나누고 올림처리
    finalnote = fullnote[0:realtime] # 올림처리한 악보를 실제 시간에 맞추어 길이 수정
    
    if search in finalnote: # 기억한 멜로디가 실제 시간에 맞춘 악보 안에 있으면
      answer.append([i, realtime, title]) # answer 안에 정보 추가
      
  answer.sort(key = lambda x:x[1], reverse=True) # answer를 realtime 기준으로 sort
  answer.append(["(None)", "(None)", "(None)"]) # answer 안에 아무 것도 없을 시에 "(None)"로 출력될 수 있도록 append

  return answer[0][2]

📑  설명

우선 전혀 깔끔한 코드는 아니다. 제출하고나니 지금 쓴 코드의 로직 자체를 더 깔끔하게 구현할 수 있었는데 라는 아쉬움이 남는다.

우선 간략한 설명을 해보겠다.

해당 문제는 아래와 같은 4가지 중심 문제가 있다..

1) 실제로 재생된 시간, 2) C#과 같은 음을 구분해야함, 3) 실제 시간에 대한 악보가 있어야함, 4) 기억한 멜로디가 있다면 재생된 시간이 가장 긴 곡을 리턴해야함

나는 먼저 1)을 구하는 time 함수를 선언하고, for 문을 통해서 musicinfos 안에 문자열을 start, end, temp_note, title로 구분했다.

그리고 2)를 해결하기위해 C#과 같은 음을 c로 치환하여 구분하였다.

3)의 문제를 해결하기 위해서는, 재생된 시간을 악보 길이로 나눈다음 올림처리하여 악보를 만들고, 그 악보를 실제 시간으로 슬라이싱하였다.

가령 실제 재생시간이 10분인데, 악보의 길이가 6이라면, 올림처리하여 악보 길이를 12로 만들고, [0:10]을 하여 자른 것이다.

마지막으로 4)의 문제는 2차원 배열 sort를 통해서, realtime 내림차순으로 정렬하였다.

더불어 일치하는 멜로디가 없을 시 "(None)"를 리턴해야하여, answer 안에 "(None)"을 넣어주었다.

그러면 return answer[0][2]할 시, 있으면 title을 일치한게 없었다면 "(None)"을 리턴할 것이다.

 

💪🏻 참고할만한 깔끔한 답안

import math

def replace_step(m):
    return m.replace('C#', 'c').replace('D#', 'd').replace('F#', 'f').replace('G#', 'g').replace('A#', 'a')

def solution(m, musicinfos):
    answer = None
    max_play_time = 0
    m = replace_step(m)

    for musicinfo in musicinfos:
        start_time, end_time, name, melody = musicinfo.split(",")
        play_time = (int(end_time[:2]) * 60 + int(end_time[3:])) - (int(start_time[:2]) * 60 + int(start_time[3:]))

        melody = replace_step(melody)
        melody_repeated_count = math.ceil(play_time / len(melody))
        melody_played = (melody * melody_repeated_count)[:play_time]
        if m in melody_played and play_time > max_play_time:
            answer = name
            max_play_time = play_time
    if not answer:
        return "(None)"
    return answer
  • 나와 문제 풀이 방법은 비슷하지만 훨씬 깔끔하고 효율적이다.