[Swift] 프로그래머스 - 주사위 게임 3 / LV.1, 68935

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

프로그래머스

 

한글로 된 문제를 간단하게 pqrt로 나타내본다면 아래와 같다.

• pppp
• pppq
• ppqq
• ppqr
• pqrt

집합을 사용해서 중복 값을 제거하는 아이디어를 떠올렸었는데
pppp -> p, pppq -> pq, ppqq -> pq, ppqr -> pqr, pqrt

pppq와 ppqq를 구분하기가 어려웠다.

 

Dictionary를 사용해서 입력 값을 key 값으로, 갯수를 value로 해봤다

 

func solution(_ a:Int, _ b:Int, _ c:Int, _ d:Int) -> Int {
     var dict = Dictionary<Int, Int>()
     
     [a, b, c, d].forEach {
         let count = dict[$0, default: 0]
         dict[$0] = count + 1
     }

    let sorted = dict.sorted(by: { $0.value > $1.value })
    
    switch sorted[0].value {
        case 1:
            return dict.sorted(by: { $0.key > $1.key }).last!.key
        case 2:
            let p = sorted[0].key
            let q = sorted[1].key
            
            if dict.count == 2 {
                return (p + q) * abs(p - q)
            } else {
                let r = sorted[2].key
                return q * r
            }
            
        case 3:
            let p = sorted[0].key
            let q = sorted[1].key
            return (10 * p + q) * (10 * p + q)
        case 4:
            return sorted[0].key * 1111
        default:
            return 0
        }
    }