[swift] 프로그래머스 - 위장

프로그래머스

---

풀이

func solution(_ clothes:[[String]]) -> Int {
    var dic = [String: Int]()
    var result = 1
    for i in clothes {
        dic[i[1]] = (dic[i[1]] ?? 0) + 1
    }
    for i in dic.values {
        result *= (i+1)
    }
    return result-1
}

func solution(_ clothes:[[String]]) -> Int {
    var dic = [String: Int]()
    for i in clothes { dic[i[1]] = (dic[i[1]] ?? 0) + 1 }
    return (dic.values.reduce(1) { $0 * ($1+1) })-1
}

 

모자: 2, 상의: 2, 신발: 1 가지의 착장이 있다고 가정하면

모자를 벗었을 때, 1번모자, 2번모자를 썻을 때, 총 3가지의 착장이 있고

상의도 벗었을 때, 1번상의, 2번상의, 총 3가지 착장이 있다.

신발은 벗었을 때, 1번신발 총 2가지 착장이 있다.

 

해당 착장들을 모두 곱하면 3 * 3 * 2 = 18가지가 나오는데

모두 벗었을 때는 제외를 해야하기 때문에 18-1 = 17이 답이 된다.

 

---