[프로그래머스] (Swift) - 위장

* 문제링크

https://programmers.co.kr/learn/courses/30/lessons/42578?language=swift

코딩테스트 연습 - 위장

 

*접근 방법 및 중요 포인트

구현보다는 경우의 수를 계산하는 방법이 헷갈리는 문제였다. 최소한 1개만 입어도 되기 때문에,

(종류별 의상수 + 1(안입는경우) * (종류별 의상수 + 1(안입는경우) * (종류별 의상수 + 1(안입는경우) .....   -1 (모두 안입으면 안되니까 빼줘야함) 이 된다.

예를들어, [썬글라스 2종류, 바지 2종류] 가 존재한다면, 경우의 수 = (2+1) * (2+1) - 1  = 8이 된다.

코드적인 부분에서는 딕셔너리로 종류별로 카운팅한다. 종류의 이름은 상관없으니 key값만 따로 뽑아서 계산해주었다.

 

* Code(Swift)

import Foundation

func solution(_ clothes:[[String]]) -> Int {
    var dict = Dictionary<String, Int>()
    
    for c in clothes{
        if dict[c[1]] == nil{
            dict[c[1]] = 1
        }else{
            dict[c[1]]! += 1
        }
    }
    var numOfClothes = Array<Int>()
    
    for (_ ,value) in dict{
        numOfClothes.append(value)
    }

    return numOfClothes.reduce(1){$0 * ($1+1)} - 1
}