2108번 '통계학'

import Foundation
 
let arrayCount = Int(readLine()!)!
var array = [Int]()
for _ in 1...arrayCount{
    array.append(Int(readLine()!)!)
}
array = array.sorted(by: <)
let average = Int((Double(array.reduce(0){ $0 + $1 }) / Double(arrayCount)).rounded())
let median = array[arrayCount / 2]
let range = array.last! - array.first!
if(arrayCount == 1){
    print(average)
    print(median)
    print(array[0])
    print(range)
    exit(0)
}
var dic = [Int: Int]()
var count = 1
for i in 0..<arrayCount - 1{
    if(array[i] == array[i+1]){
        count += 1
        dic[array[i]] = count
    }
    else {
        dic[array[i]] = count
        count = 1
    }
}
if(array[arrayCount - 2] != array[arrayCount - 1]){
    dic[array[arrayCount - 1]] = 1
}
var sortedDic = dic.sorted(by: { (first, second) -> Bool in
    return first.key < second.key
})
let max = {() -> Int in
    var values = [Int]()
    for i in sortedDic{
        values.append(i.value)
    }
    return values.max()!
}()
sortedDic = sortedDic.filter { (key, value) -> Bool in
    return value == max
}
let mode: Int
if(sortedDic.count == 1){
    mode = (sortedDic.first?.key)!
} else {
    sortedDic = sortedDic.sorted(by: { (first, second) -> Bool in
        return first.key < second.key
    })
    mode = sortedDic[1].key
}
print(average)
print(median)
print(mode)
print(range)

코드가 너무 더럽다...

어느 정도 푼 다음 채점 결과를 기다렸는데 100%에서 런타임 오류가 나길래 생각해 봤는데

N이 1일 때 최빈값을 구하는 코드로 들어가면 잘못된 인덱스를 참조하는 것이 되므로 해당 경우에 따로 처리해 준다.