[Algorithm] 베스트앨범

[hwangJi-dev]

💬 문제

https://school.programmers.co.kr/learn/courses/30/lessons/42579

💬 Idea

1. struct 구조체를 만들어 음악 정보를 저장해준다.
2. genreDict를 만들어 재생횟수를 저장해준다
3. 최대 재생횟수를 가지는 장르 순위대로 베스트앨범을 채운다.
   1. 가장 많이 재생된 장르
   2. 해당 장르 중 가장 많이 재생된 노래
      1. 재생횟수가 같다면 고유번호 낮은 순으로 정렬

💬 풀이

import Foundation

struct Music {
    var genre: String
    var playCnt: Int
    var index: Int
}

func solution(_ genres:[String], _ plays:[Int]) -> [Int] {
    var genreDict: [String: Int] = [:]
    var musics: [Music] = []
    var album: [Int] = []
    
    for (idx, genre) in genres.enumerated() {
        musics.append(Music(genre: genre, playCnt: plays[idx], index: idx))
        if genreDict[genre] == nil {
            genreDict[genre] = plays[idx]
        } else {
            genreDict[genre]! += plays[idx]
        }
    }
    
    musics = musics.sorted(by: {
        if $0.playCnt == $1.playCnt {
            return $0.index < $1.index
        } else {
            return $0.playCnt > $1.playCnt
        }
    })
    
    for i in genreDict.sorted(by: { $0.value > $1.value }) {
        let music = musics.filter({ $0.genre == i.key }).map({ $0.index })
        if music.count >= 2 {
            album += music[0...1]
        } else {
            album.append(music[0])
        }
    }
    
    return album
}

소요시간 : 23분

💬 더 나은 방법?

• 구조체 배열을 담는 변수를 제거하고 Dictionary에 튜플 형식으로 응용하여 사용

struct Music {
    var playCnt: Int
    var index: Int
}

func solution(genres:[String], plays:[Int]) -> [Int] {
    var genreDict: [String: ([Music], Int)] = [:]
    var album: [Int] = []
    
    for (idx, genre) in genres.enumerated() {
        if genreDict[genre] != nil {
            genreDict[genre]!.0.append(Music(playCnt: plays[idx], index: idx))
            genreDict[genre]!.1 += plays[idx]
        } else {
            genreDict[genre] = ([Music(playCnt: plays[idx], index: idx)], plays[idx])
        }
    }
    
    for i in genreDict.sorted(by: { $0.value.1 > $1.value.1 }) {
        let music = i.value.0.sorted(by: {
            return ($0.playCnt == $1.playCnt) ? $0.index < $1.index : $0.playCnt > $1.playCnt
        }).map({ $0.index })
        album += music.count >= 2 ? music[0...1] : [music[0]]
    }
    
    return album
}