코니는 매일 다른 옷을 조합하여 입는것을 좋아합니다.
예를 들어 코니가 가진 옷이 아래와 같고, 오늘 코니가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야합니다.
종류 | 이름 |
---|---|
얼굴 | 동그란 안경, 검정 선글라스 |
상의 | 파란색 티셔츠 |
하의 | 청바지 |
겉옷 | 긴 코트 |
코니가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.
종류에 따라서 값들을 정리해야 된다.
(모자의 갯수 + 1) * (상의의 갯수 + 1) \* (하의의 갯수 + 1) - 1
이다.이러한 동작을 하는 코드를 작성하면 문제 해결!
import Foundation
func solution(_ clothes:[[String]]) -> Int {
var sorted: [String: [String]] = [:]
var count = 0
var combination = 1
for clothe in clothes {
if sorted.keys.contains(clothe[1]) {
sorted[clothe[1]]?.append(clothe[0])
} else {
sorted[clothe[1]] = [clothe[0]]
}
}
for part in sorted {
combination *= (part.value.count + 1)
count += part.value.count
}
return combination - 1
}