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

Source Code

---

	import Foundation
	func solution(_ genres:[String], _ plays:[Int]) -> [Int] {
	//장르와 장르의 인덱스를 가질 딕셔너리
	var genresAndPlays = [String:[Int]]()
	//장르와 장르 플레이 토탈의 수를 가질 딕셔너리들을 가지고 있는 배열
	var total = [[String:Int]]()
	//정답 배열
	var answer = [Int]()
	//play안을 탐색해줍니다.
	for (current,play) in plays.enumerated() {
	//장르안에 현재 인덱스를 key로 합니다.
	let key = genres[current]
	//만약 genresAndPlays가 nil이라면
	if genresAndPlays[key] == nil {
	//현재 genresAndPlays에 인덱스를 추가해줍니다.
	genresAndPlays[key] = [current]
	//total에는 새로운 key와 play시간을 가지는 딕셔너리를 넣어줍니다.
	total.append([key:play])
	continue
	}
	//genresAndPlays를 탐색해줍니다.
	for (j,index) in genresAndPlays[key]!.enumerated() {
	//만약 현재 인덱스의 play가 genresAndPlays[key]안에 있는 play보다 크다면
	if plays[index] < plays[current] {
	//그 자리에 삽입해줍니다.
	genresAndPlays[key]?.insert(current, at: j)
	//만약 genresAndPlays[key]의 수가 2보다 크다면
	if genresAndPlays[key]!.count > 2 {
	//가장 마지막 인덱스를 삭제해줍니다.
	genresAndPlays[key]?.removeLast()
	}
	break
	//만약 현재 인덱스의 play가 genresAndPlays[key]안에 있는 play보다 작거나 같다면
	}else{
	//만약 genresAndPlays[key]가 2보다 작다면
	if genresAndPlays[key]!.count < 2 {
	//가장 뒤에 넣어줍니다.
	genresAndPlays[key]?.append(current)
	}
	}
	}
	//total의 딕셔너리 중 현재 key를 가지고 있는 딕셔너리가 있는 인덱스를 찾아줍니다.
	let tIndex = total.firstIndex { (dictionary) -> Bool in
	dictionary.first?.key == key
	}
	//찾은 인덱스에서 key가 있는 곳에 play를 더해줍니다.
	total[tIndex!][key]! += play
	}
	//total은 딕셔너리를 하나씩 가지고 있는 배열이므로 각 딕셔너리의 밸류값을 기준으로 큰 값으로 정렬해줍니다.
	let song = total.sorted {$0.first!.value > $1.first!.value}
	//토탈수대로 반복해줍니다.
	for i in 0...total.count-1 {
	//answer에 genresAndPlays의 song의 첫번째 key가 가지고 있는 인덱스들을 넣어줍니다.
	answer.append(contentsOf: genresAndPlays[song[i].first!.key]!)
	}
	//인덱스를 반환해줍니다.
	return answer
	}

view raw 베스트앨범.swift hosted with ❤ by GitHub

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

테스트 케이스 효율성 비교

 

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

 

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