[programmers/Swift] 위장

programmers.co.kr - 코딩테스트연습 - Lv.2 - 해시 - 위장

코딩테스트 연습 - 위장

Constraints :

• clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
• 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
• 같은 이름을 가진 의상은 존재하지 않습니다.
• clothes의 모든 원소는 문자열로 이루어져 있습니다.
• 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
• 스파이는 하루에 최소 한 개의 의상은 입습니다.

Solution.swift :

//
//  Created by Yongwoo Marco on 2022/05/10.
//  Copyright © 2022 Yongwoo Marco Kim. All rights reserved.
//

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

print(solution([["yellowhat", "headgear"], 
				["bluesunglasses", "eyewear"], 
				["green_turban", "headgear"]]))
// 5
print(solution([["crowmask", "face"], 
				["bluesunglasses", "face"], 
				["smoky_makeup", "face"]]))
// 3

How I tried this :
같은 이름을 가진 의상은 존재하지 않기 때문에 각 부위별 갯수만 세고,

((가지 수 + 안 입을 수도 있는 점) * 부위 수) -  하나도 안입은 경우

형태로 답을 구함

문제에 관한 모든 저작권 : https://programmers.co.kr/