프로그래머스 베스트 앨범 Swift

Source Code

---

Solution

---

해당 문제는 해쉬문제였다.(Swift로 따지면 Dictionary를 사용해 풀어야하는 문제)

 

1. 각 장르를 키값으로 장르가 있는 인덱스를 밸류값으로 딕셔너리를 만들어서 담아준다.

 

2. 1에서 담은 인덱스들은 플레이 값을 기준으로 가장 큰 2개의 인덱스만 담아준다.

 

3. 각 장르를 키값으로 하고 각 장르의 인덱스가 있는 플레이 값의 총합을 밸류값으로 갖는 딕셔너리를 만들어서 담아준다.

 

4. 총합이 가장 높은 순으로 딕셔너리를 정렬해준뒤 인덱스를 담고 있는 밸류값들을 각 장르별로 모아준 다음 반환해준다.

 

Problem

---

 

코딩테스트 연습 - 베스트앨범

새롭게 알게 된 것

 

딕셔너리를 담은배열은 인덱스가 없고 순서도 랜덤으로 배치된다.

 

고로 배열에 딕셔너리를 담아서 정렬을 해주어야 된다.

 

다른 사람 풀이

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33

import Foundation     class Album {     var index : Int     var playTime : Int       init(_ index : Int, _ playTime : Int) {           self.index = index         self.playTime = playTime     } }   func solution(_ genres:[String], _ plays:[Int]) -> [Int] {       var albumDic = [String : [Album]]()       plays     .enumerated()     .map     { (genres[$0], Album($0, $1)) }     .forEach { albumDic[$0, default: [Album]() ].append($1) }       return albumDic     .sorted {         $0.value.reduce(0, { $0 + $1.playTime }) > $1.value.reduce(0, { $0 + $1.playTime })     }     .map { _, albums -> [Int] in           guard albums.count > 1 else { return [albums[0].index] }           let sorted = albums.sorted(by: { $0.playTime > $1.playTime} )           return [sorted[0].index, sorted[1].index]}     .reduce(into: [Int]()) { $0.append(contentsOf : $1) } } Colored by Color Scripter

테스트 케이스 효율성 비교

 

좌측 내 코드 우측 다른 사람 코드

 

그래도 이번에는 비슷비슷하게 나왔다.