[programmers] 3차 방금 그 곡 python

[프로그래머스][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개 이하로 구성되어 있다.
  출력 형식
  조건과 일치하는 음악 제목을 출력한다.

풀이과정

이 문제는 정확히 무슨 뜻인지를 모르겠어서 풀이를 할 수가 없다.. 문제에서 시키는 대로 했을 때 실제 채점에서 틀리는 부분이 너무 많아서 프로그래머스의 질문하기에서 다른 분들이 주시는 테스트케이스에 맞춰가는 식으로 풀었는데도... 내 코드에서는 30번은 솔브하지 못했다!!!!!!!!!!!!!!!! 완전 😱

• 추가 테스트 케이스

  print(solution("A#",["23:56,00:00,HAPPY,B#A#"])) #happy
  print(solution("CDEFGAC",["12:00,12:06,HELLO,CDEFGA"])) #none
  print(solution("CCB",["03:00,03:10,FOO,CCB#CCB", "04:00,04:08,BAR,ABC"])) #foo

소스코드

소요시간 01:53:08

정답코드

import datetime
def solution(m, musicinfos):
    answer = ['', 0] #타이틀, 재생시간

    # #이 들어있는 것은 하나로 합쳐준다.
    m_list = []
    for m1 in m:
        if m1 == '#':
            m_list.append(m_list.pop() + m1)
        else:
            m_list.append(m1)
    mm = ''.join(m_list)

    musicinfos = [x.split(',') for x in musicinfos]

    for info in musicinfos: #info = [시작시간, 끝시간, 제목, 멜로디]
        start = datetime.datetime.strptime(info[0],'%H:%M')
        end = datetime.datetime.strptime(info[1], '%H:%M')
        dt = ((end-start).seconds) // 60

        melody = list()
        for mu in info[3]:
            if mu == '#':
                melody.append(melody.pop() + mu)
            else:
                melody.append(mu)
        else:
            # 멜로디의 길이가 주어진 시간보다 길다면
            if dt < len(melody):
                melody = melody[:dt]
            else:
                melody = melody * (dt // len(melody)) + melody[:(dt % len(melody))]
            # m의 길이만큼 melody를 쪼개서, m과 동일한지 확인
            for i in range(len(melody) - len(m_list) + 1):
                # melody를 m의 길이만큼 쪼갠 것과 m이 같고, 재생 시간이 길다면
                if ''.join(melody[i:i+len(m_list)]) == mm and dt > answer[1]:
                    answer = [info[2], dt]

    if answer[1]:
        return answer[0]
    else:
        return '(None)'

2시간 개고생했는데 30번에서 짤당하는 코드

import datetime
melody = {'A': 'A', 'A#': 'B', 'B': 'C', 'C': 'D', 'C#': 'E', 'D': 'F', 'D#': 'G', 'E': 'H', 'F': 'I', 'F#': 'J','G': 'K', 'G#': 'L'}
def calc_time(start, end):
    start = datetime.datetime.strptime(start, '%H:%M')
    end = datetime.datetime.strptime(end, '%H:%M')
    return (end-start).seconds // 60

def replace(sheet):
    result = ''
    for i in range(len(sheet)):
        word = ''
        if sheet[i] == '#':
            result = result[:-1]
            if sheet[i - 1] + sheet[i] in melody.keys():
                word = melody[sheet[i - 1] + sheet[i]]
        else:
            word = melody[sheet[i]]
        result += word
    return result

def make_sheet(playing_time, sheet):
    if len(sheet) > playing_time:  # 주어진 재생시간보다 곡 길이가 긴 경우
        # return sheet[:playing_time+1]
        return sheet
    else:
        multiple = playing_time // len(sheet)
        remainder = playing_time % len(sheet)
        result = ''
        for _ in range(multiple):
            result += sheet
        result += sheet[:remainder]
        return result


def get_that_song(answers, m):
    result = []
    for i in range(len(answers)):
        sheet = answers[i][0]
        if m in sheet:
            result.append(answers[i])
    return result


def solution(m, musicinfos):
    answers = []

    # melody dic에 있는대로 멜로디를 변경해줌.
    m = replace(m)
    musicinfo = []
    for info in musicinfos:
        music = list(info.split(','))
        music[3] = replace(music[3])
        musicinfo.append(music)
    for i in range(len(musicinfo)):
        start_time, end_time, title, sheet = musicinfo[i]
        playing_time = calc_time(start_time, end_time)
        answers.append([make_sheet(playing_time, sheet), title, playing_time, i])
        # 멜로디, 제목, 재생 시간을 추가

    answers = get_that_song(answers, m)
    answers = sorted(answers, key=lambda x: (-x[2], x[3]))
    if len(answers) == 0:
        return '(None)'
    else:
        return answers[0][1]