coding test

[파이썬, Java] [3차] 방금그곡

잔망루피 2021. 9. 6. 18:48

문제 설명

방금그곡

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

해설 보러가기

.

 

💛 나의 풀이

 

def replace_song(song):
    song = song.replace("C#", "c")
    song = song.replace("D#", "d")
    song = song.replace("F#", "f")
    song = song.replace("G#", "g")
    song = song.replace("A#", "a")
    return song

# 재생시간 만큼 노래 만들기
def song(m, musicinfos):
    cnt = musicinfos[4]  # 재생시간
    len_song = len(musicinfos[3])  # 악보 길이
    song = ""  # 실제 재생된 음악
    time = 0
    i = 0
    while True:
        if time == cnt:
            break
        song += musicinfos[3][i % len_song]
        time += 1
        i += 1

    return song


def compare(m, song):
    if m not in song :
        return False
    return True


def solution(m, musicinfos):
    answer = ''  # 노래 제목을 담을 문자열 변수
    record = musicinfos  # 2차원 리스트
    m = replace_song(m)		# '#'이 붙어 있으면 소문자로 치환 

    for i in range(len(musicinfos)):
        record[i] = musicinfos[i].split(",")  # ,를 기준으로 문자열 나누고 2차원배열에 담기
        record[i][3] = replace_song(record[i][3])	# '#'이 붙어 있으면 소문자로 치환 

        # 시간 변환
        start = record[i][0].split(":")
        end = record[i][1].split(":")
        hour = int(end[0]) - int(start[0])
        minute = int(end[1]) - int(start[1])
        time = hour * 60 + minute
        record[i].append(time)  # 재생시간 추가

    record.sort(key=lambda x: x[4], reverse=True)	# 재생시간 순으로 역순 정렬

    for i in range(len(musicinfos)):
        result = song(m, record[i])  # 재생시간만큼 노래 재생
        if compare(m, result):
            answer = musicinfos[i][2]
            break

    return answer if answer else "(None)"

왜 몇 개가 틀리나 했더니 " (None) "이 아니었다... 내 눈에는 이렇게 보여서 했더니 ㅠ

"#"이 붙은 것은 어떻게 처리하나 했더니 치환하면 되는 거였다!!

나는 뒤에 또 #있나 확인하는 if문을 썼다가 위 방법을 알고 수정함.

재생 시간은 ':'를 기준으로 나눈 후 시, 분끼리 뺀다.

시는 60을 곱해서 분으로 환산했다. 처음에는 이 과정을 까먹고 안 했었다 🤣

네오가 기억한 멜로디 m이 재생된 노래 song에 있으면 answer에 담고 break한다.

아래는 통과하기 위한 흔적들

 

 

class Solution {
    static String[] split_info;
    public String replace(String str){
        str=str.replace("C#", "c");
        str=str.replace("D#", "d");
        str=str.replace("F#", "f");
        str=str.replace("G#", "g");
        str=str.replace("A#", "a");
        return str;
    }

    public String solution(String m, String[] musicinfos) {
        String[] answer = {"(None)", "0"};
        m=replace(m);
        for(int i=0; i<musicinfos.length; i++){
            musicinfos[i]=replace(musicinfos[i]);
            split_info=musicinfos[i].split(",");
            int hour=Integer.parseInt(split_info[1].substring(0, 2))-Integer.parseInt(split_info[0].substring(0, 2));
            int minute=Integer.parseInt(split_info[1].substring(3))-Integer.parseInt(split_info[0].substring(3));
            int time=hour*60+minute;        // 재생시간
            int len=split_info[3].length();
            String song="";
            for(int j=0; j<time/len; j++){
                song+=split_info[3].concat(split_info[3]);
            }
            song+=split_info[3].substring(0, time%len);
            if (song.contains(m) && time > Integer.parseInt(answer[1])) {
                answer[0]=split_info[2];    // 제목
                answer[1]=Integer.toString(time);
            }

        }
        return answer[0];
    }
}

다른 사람의 풀이를 공부하면서 배운 정렬 대신 갱신하는 방법을 사용했다.

재생시간 time이 주어진 악보 길이보다 길어서 n회 반복해야 할 때 concat으로 이어준다.

반복이 1회 미만일 경우 %연산자를 이용해 나온 나머지만큼 잘라준다.

 

# 실패
def replace_song(song) :
    song=song.replace("C#", "c")
    song=song.replace("D#", "d")
    song=song.replace("F#", "f")
    song=song.replace("G#", "g")
    song=song.replace("A#", "a")
    return song


def song(m, musicinfos):
    cnt = musicinfos[4]     # 재생시간
    len_song=len(musicinfos[3])     # 악보 길이
    song = ""               # 실제 재생된 음악
    time = 0
    i=0
    while True:
        if time == cnt:
            break
        song += musicinfos[3][i%len_song]
        time += 1
        i+=1

    return song


def compare(m, song) :
    try :       # 런타임 에러 해결
        start=song.index(m[0])
    except :
        return False
    tmp=song[start:]+song[0:start]
    if len(tmp) >= len(m) :
        if m not in tmp :
            return False
    elif tmp not in m :
        return False
        
    return True


def solution(m, musicinfos):
    answer = ''  # 노래 제목을 담을 문자열 변수
    record = musicinfos  # 2차원배열
    m=replace_song(m)

    for i in range(len(musicinfos)):
        record[i] = musicinfos[i].split(",")  # ,를 기준으로 문자열 나누고 2차원배열에 담기
        record[i][3]=replace_song(record[i][3])
        
        # 시간 변환
        start=record[i][0].split(":")
        end=record[i][1].split(":")
        hour=int(end[0])-int(start[0])
        minute=int(end[1])-int(start[1])
        time=hour*60+minute
        record[i].append(time)  # 시간 추가

    record.sort(key=lambda x:x[4], reverse=True)

    for i in range(len(musicinfos)):
        result = song(m, record[i])  # 재생시간만큼 노래 재생
        if compare(m, result) :
            answer=musicinfos[i][2]
            break

    return answer if answer else " (None) "

방송된 곡에서 네오가 기억한 멜로디 m의 첫 문자 위치를 찾는다.

tmp에 song[첫 문자 위치 : 끝]+[0:첫 문자 위치 전]을 넣었다.

tmp와 m 중 길이가 더 긴 쪽이 기준이 된다.

제출하니 86.7정도 받았던 것 같다.

내가 생각한 이 아이디어 자체가 좀 부족한 부분이 있다.

문제에서 노래는 반드시 처음부터 재생된다고 했는데 이 부분을 어겼구나...

 

# 실패
def song(m, musicinfos):
    cnt = musicinfos[4]     # 재생시간
    len_song=len(musicinfos[3])     # 악보 길이
    song = ""               # 실제 재생된 음악
    time = 0

    for i in range(1440):
        if time == cnt:
            break
        if musicinfos[3][i%len_song] != "#":
            song += musicinfos[3][i%len_song]
            time += 1
        else:
            song += musicinfos[3][i%len_song]

    return song


def compare(m, song) :
    start=song.index(m[0])
    tmp=song[start:]+song[0:start]
    
    if tmp[:len(m)] != m :
        return False
    elif tmp[len(m)] == "#" :
        return False
    
    return True


def solution(m, musicinfos):
    answer = ''  # 노래 제목을 담을 문자열 변수
    record = musicinfos  # 2차원배열
    
    for i in range(len(musicinfos)):
        record[i] = musicinfos[i].split(",")  # ,를 기준으로 문자열 나누고 2차원배열에 담기
        for j in range(2):
            record[i][j] = record[i][j].replace(":", "")  # :없애기
        length = int(record[i][1]) - int(record[i][0])  # 시간차 구하기
        record[i].append(length)  # 시간 추가

    record.sort(key=lambda x:x[4], reverse=True)

    for i in range(len(musicinfos)):
        result = song(m, record[i])  # 재생시간만큼 노래 재생
        if compare(m, result) :
            answer=musicinfos[i][2]

    return answer if answer else (None)

index 메소드는 문자열에서 찾으려는 문자를 찾지 못하면 에러가 발생한다.

 

 

🦄 다른 사람 풀이

# https://eda-ai-lab.tistory.com/506
# #이 붙은 음표는 소문자로 변경
def changecode(music_) :
    music_=music_.replace('C#', 'c')
    music_=music_.replace('D#', 'd')
    music_=music_.replace('F#', 'f')
    music_=music_.replace('G#', 'g')
    music_=music_.replace('A#', 'a')
    return music_

def caltime(musicinfo_) :
    starttime=musicinfo_[0]
    endtime=musicinfo_[1]
    hour=1*(int(endtime.split(':')[0])-int(starttime.split(':')[0]))
    if hour == 0 :
        total=int(endtime.split(':')[1])-int(starttime.split(':')[1])
    else :
        total=60*hour+int(endtime.split(':')[1])-int(starttime.split(':')[1])
    
    return total

def solution(m, musicinfos) :
    answer=[]
    m=changecode(m)
    for idx, musicinfo in enumerate(musicinfos) :
        musicinfo=changecode(musicinfo)
        musicinfo=musicinfo.split(',')
        time=caltime(musicinfo)
        
        # 길이가 시간보다 더 긴 경우
        if len(musicinfo[3]) >= time :
            melody=musicinfo[3][0:time]
        else :
            # 시간을 계산해서 몫과 나머지로 계산
            # 다른 사람 풀이 : q, r=divmod(time, len(musicinfo[3]))
            a=time//len(musicinfo[3])
            b=time%len(musicinfo[3])
            melody=musicinfo[3]*a+musicinfo[3][0:b]
        # 노래가 melody에 포함되면 정답후보에 저장
        if m in melody :
            answer.append([idx, time, musicinfo[2]])
    # 정답이 있는 경우
    if len(answer) != 0 :
        # time -> index 기준으로 정렬
        answer=sorted(answer, key=lambda x : (-x[1], x[0]))
        return answer[0][2]
    #정답이 없는 경우
    return "(None)"

나는 정렬 후 작업을 했는데 이 풀이는 정답이 될 수 있는 곡들을 정렬 후 가장 앞에 있는 값을 반환한다.

 

 

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

def solution(m, musicinfos) :
    answer=[0, '(None)']    # time_len, title
    m=shap_to_lower(m)
    for info in musicinfos :
        split_info=info.split(',')
        time_length=(int(split_info[1][:2])-int(split_info[0][:2]))*60+int(split_info[1][-2:])-int(split_info[0][-2:])
        title=split_info[2]
        part_notes=shap_to_lower(split_info[-1])
        full_notes=part_notes*(time_length//len(part_notes))+part_notes[:time_length%len(part_notes)]
        if m in full_notes and time_length>answer[0] :
            answer=[time_length, title]
    return answer[-1]

재생시간을 계산할 때 슬라이싱으로 :을 제외한 시, 분끼리 계산한다.

정렬 대신 재생시간이 더 긴 노래가 있으면 answer을 갱신한다.

 

 

 

문제 출처 👉 프로그래머스

반응형

'coding test' 카테고리의 다른 글

[파이썬] 1021. 회전하는 큐  (0) 2021.09.14
[파이썬, Java] [3차] 압축  (0) 2021.09.06
[Java] 2023. 신기한 소수  (0) 2021.09.01
[Java, Python] 1759. 암호 만들기  (0) 2021.08.31
[Java] 6603. 로또  (0) 2021.08.31