[Swift] 프로그래머스 LV.0 최빈값 구하기

프로그래머스 LV.0 모음

최빈값 구하기

문제 설명

최빈값은 주어진 값 중에서 가장 자주 나오는 값을 의미합니다. 정수 배열 array가 매개변수로 주어질 때, 최빈값을 return 하도록 solution 함수를 완성해보세요. 최빈값이 여러 개면 -1을 return 합니다.

제한 사항

• 0 < array의 길이 < 100
• 0 ≤ array의 원소 < 1000

입출력 예

array	result
[1, 2, 3, 3, 3, 4]	3
[1, 1, 2, 2]	-1
[1]	1

입출력 예 설명

입출력 예 #1

• [1, 2, 3, 3, 3, 4]에서 1은 1개 2는 1개 3은 3개 4는 1개로 최빈값은 3입니다.

입출력 예 #2

• [1, 1, 2, 2]에서 1은 2개 2는 2개로 최빈값이 1, 2입니다. 최빈값이 여러 개이므로 -1을 return 합니다.

입출력 예 #3

• [1]에는 1만 있으므로 최빈값은 1입니다.

---

제출

import Foundation

func solution(_ array:[Int]) -> Int {
    var dict = [Int: Int]()

    for a in array{
        dict[a, default: 0] += 1
    }

    let max = dict.max{$0.value <= $1.value}!

    return dict.values.filter{$0 == max.value}.count > 1 ? -1 : max.key
}

딕셔너리를 선언하여 배열의 요소를 key 값으로, 각 요소의 개수를 value 값으로 저장한다.
딕셔너리에서 value가 가장 큰 값을 찾아 가장 큰 value를 가진 key가 여러 개라면 -1을, 아니라면 key를 반환한다.

 

다른 풀이

func solution(_ array: [Int]) -> Int {
    let sorted = Dictionary(grouping: array) { $0 }.sorted { $0.value.count > $1.value.count }
    return sorted.count > 1 && sorted[0].value.count == sorted[1].value.count ? -1 : sorted[0].key
}

배열을 딕셔너리로 그룹핑한 후 value의 개수에 따라 내림차순으로 정렬한다.
정렬한 배열에서 첫 번째 배열의 개수와 두 번째 배열의 개수가 같다면 -1을, 아니라면 첫 번째 배열의 키값을 반환한다.