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

코드

import Foundation

func solution(_ clothes:[[String]]) -> Int {
	// 옷의 종류와 갯수 저장
    var dic = [String : Int]()
    var sum = 1
    
    for i in 0..<clothes.count {
        dic[clothes[i][1]] = (dic[clothes[i][1]] ?? 0) + 1
    }
    
    for (key, value) in dic {
    	// 각각 옷의 갯수에 +1을 하고 곱한다
        sum *= (value+1)
    }
    
    if dic.count == 1 {
        return clothes.count
    } else {
    	// 아무것도 입지 않는 경우는 없기에 -1을 해준다
        return sum - 1
    }
}

회고

• 코드짜는 것은 생각보다 쉬웠다
• 경우의 수 수학 공식이 생각이 나지 않아 검색을 하였다
• 딕셔너리를 어느정도 이제 사용할 수 있는 것 같아 기분이 좋다