[Swift - 프로그래머스] [3차] 방금그곡

🕵🏻  문제 해석

---

네오는 라디오에서 흘러나온 노래의 멜로디의 일부분을 기억하고 방금 그 곡이라는 서비스를 이용하여 해당 노래의 제목을 찾으려고 합니다. 이때 라디오 채널의 재생된 노래의 제목, 재생 시각, 전체 멜로디 정보를 이용하여 문제를 해결해보려 합니다.

주어지는 정보와 출력에는 몇 가지 규칙이 존재합니다.

• 네오가 기억한 멜로디와 악보에 사용되는 음은 C, C#, D, D#, E, F, F#, G, G#, A, A#, B 12개이다.
• 각 음은 1분에 1개씩 재생된다. 음악은 반드시 처음부터 재생되며 음악 길이보다 재생된 시간이 길 때는 음악이 끊김 없이 처음부터 반복해서 재생된다. 음악 길이보다 재생된 시간이 짧을 때는 처음부터 재생 시간만큼만 재생된다.
• 음악이 00:00을 넘겨서까지 재생되는 일은 없다.
• 조건이 일치하는 음악이 여러 개일 때에는 라디오에서 재생된 시간이 제일 긴 음악 제목을 반환한다. 재생된 시간도 같을 경우 먼저 입력된 음악 제목을 반환한다.
• 조건이 일치하는 음악이 없을 때에는 “(None)”을 반환한다.
• m은 음 1개 이상 1439개 이하로 구성되어 있다.
• musicinfos는 100개 이하의 곡 정보를 담고 있는 배열로, 각각의 곡 정보는 음악이 시작한 시각, 끝난 시각, 음악 제목, 악보 정보가 ', '로 구분된 문자열이다.
• 음악의 시작 시각과 끝난 시각은 24시간 HH:MM 형식이다.
• 음악 제목은 ', ' 이외의 출력 가능한 문자로 표현된 길이 1 이상 64 이하의 문자열이다.
• 악보 정보는 음 1개 이상 1439개 이하로 구성되어 있다.

 

저는 먼저 주어진 음악의 멜로디를 재생시간에 맞게 끔 반복 재생한 문자열을 구하고 그 문자열이 네오가 기억한 멜로디 문자열을 포함하고 있으면 정답을 리턴하는 방식으로 접근했습니다.

그 과정에서 몇 가지 문제가 있었는데, 그중에 하나가 C와 C#를 구분해주어야 한다는 점이었습니다. 단순히 contains 메서드를 사용하면 기억했던 멜로디가 "ABC"이고 재생된 노래들이 "ABC", "ABC#" 일 때 두 노래 모두 매칭이 되어 정확한 답을 구할 수 없기 때문이었습니다.

이를 해결해주기 위해 C#을 소문자 c로 변환해주어 구분이 되도록 해결했습니다. 그 후 매칭이 되는 노래들을 배열에 담아서 재생시간, 입력 순서에 따라 정렬을 해준 뒤 정답을 구해보았습니다.

📦  문제 풀이 & 코드

---

import Foundation

struct MusicInfo {
    let title: String
    let playtime: Int
    let score: String
    var playedMelody: String {
        let start = score.startIndex
        
        // 재생시간이 해당곡의 멜로디 보다 짧은 경우
        if playtime < score.count {
            return String(score[start...score.index(start, offsetBy: playtime)])
        }
        
        // 그렇지 않을 경우 나머지 연산을 통해 재생시간에 맞춰 재생된 멜로디 문자열을 반환
        return [String](repeating: score, count: playtime / score.count).joined()
        + String(score[start...score.index(start, offsetBy: playtime % score.count)])
    }
    
    init(_ title: String, _ playtime: Int, _ score: String) {
        self.title = title
        self.playtime = playtime
        self.score = score
    }
}

func solution(_ m:String, _ musicinfos:[String]) -> String {
    
    // 매칭된 음악 리스트를 저장할 배열
    var matchedMusicList: [(title: String, playtime: Int)] = []
    let formatter =  DateFormatter()
    formatter.dateFormat = "HH:mm"
    
    // DateFormatter를 이용하여 재생시간을 분 단위로 구하는 함수
    
    func calculatePlaytime(_ start: String, _ end: String) -> Int {
        let startDate = formatter.date(from: start)!
        let endDate = formatter.date(from: end)!
        
        return Int(endDate.timeIntervalSince(startDate) / 60)
    }
    
    // 입력된 멜로디 문자열의 #노트들을 소문자로 변환하는 함수
    
    func convertSharpToLowcase(_ score: String) -> String {
        var convertedScore = score
        let notes = [("C#","c"), ("A#","a"), ("D#","d"), ("F#","f"), ("G#","g")]
        
        for note in notes {
            convertedScore = convertedScore.replacingOccurrences(of: note.0, with: note.1)
        }
        
        return convertedScore
    }
    
    for musicinfo in musicinfos {
        let components = musicinfo.components(separatedBy: ",")
        let convertedScore =  convertSharpToLowcase(components[3])
        let music = MusicInfo(components[2], calculatePlaytime(components[0], components[1]), convertedScore)
        
        // 매칭된 음악을 리스트에 저장
        if music.playedMelody.contains(convertSharpToLowcase(m)) {
            matchedMusicList.append((music.title, music.playtime))
        }
    }
    
    // 매칭된 음악이 없을 경우 "(None)" 리턴
    
    if matchedMusicList.isEmpty {
        return "(None)"
    }
    
    // 재생 시간, 입력순서에 맞춰 배열을 정렬, 재생시간이 긴 음악이 앞에 오도록 정렬
    
    let sorted = matchedMusicList.enumerated().sorted {
        if $0.element.playtime == $1.element.playtime {
            return $0.offset < $1.offset
        }
        return $0.element.playtime > $1.element.playtime
    }
    
    // 배열의 첫 번째 원소가 정답
    
    return sorted.first!.element.title
}