[Programmers] 위장

문제링크 👉 https://programmers.co.kr/learn/courses/30/lessons/42578?language=swift

 

문제 이해 

스파이들이 매일 다른 옷으로 위장하는데 그 경우의 수를 구하는 문제입니당

문제 풀이

문제 입출력 예 1번을 보면, 

headgear로는 "yellowhat", "green_turban" 2개가 있고

eyewear로는 "bluesunglasses" 1개가 있습니다. 

그래서 스파이가 옷을 입는 경우의 수로는 

1. yellow_hat
2. blue_sunglasses
3. green_turban
4. yellow_hat + blue_sunglasses
5. green_turban + blue_sunglasses

이렇게 5개를 생각할 수 있어요. 옷의 종류는 한개씩밖에 못 입으니까 경우의 수 조합 문제로 생각해보면, 

 

headgear : 안 입는경우 + yellowhat 입는 경우 + green_turnban 입는 경우 총 3개의 경우의 수가 있고 

eyewear : 안 입는 경우 + bluesunglasses 입는 경우 총 2개의 경우의 수가 있어요.

 

그래서 3 x 2 = 6 인데,

스파이는 하루에 최소 한 개의 의상은 입는다는 조건 때문에 전부 안 입는 경우를 빼주어야 합니다!

 

저는 옷 배열을 받자마자 딕셔너리로 옷종류의 수를 저장했습니다. 

 

코드

import Foundation

func solution(_ clothes:[[String]]) -> Int {
    var clothesKind: [String: Int] = [:]
    
    clothes.forEach {
        var count: Int = clothesKind[$0[1]] ?? 1
        count += 1
        clothesKind[$0[1]] = count
    }
    
    var result: Int = 1
    for key in clothesKind.keys {
        result *= clothesKind[key]!
    }
    return result - 1
}