패션왕 신해빈 (백준 9375번)

문제

해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?

입력

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.

각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.

문제 풀이 코드🧑🏻‍💻

해당 문제는 알몸(의상을 0개 입은 상태)이 아닌 상태의 경우의 수를 구하는 문제입니다.

같은 종류의 의상은 하나만 입을 수 있다는 조건하에 조합할 수 있는 모든 경우의 조합 수에서 알몸의 경우만 빼면 원하는 결과를 얻을 수 있습니다.

또한 같은 이름을 가진 의상은 존재하지 않지만(key), 같은 종류의 의상은 중복을 허용(value)한다는 특징을 활용하여 코드 설계 시 배열(Array)이 아닌 딕셔너리(Dictionary)를 활용하였습니다.

let testCase = Int(readLine()!)!
var wearDic = [String:Int]()
var resultArray = [Int]()
for _ in 0..<testCase{
    let n = Int(readLine()!)!
    for _ in 0..<n{
        let wearKey = readLine()!.split(separator: " ").map{String($0)}[1]
        //Swift에는 증감 연산자(++)가 없기 때문에 if문 활용
        if wearDic[wearKey] != nil{  //딕셔너리에 값이 있으면 해당 값에 1추가
            wearDic[wearKey]! += 1
        }
        else{  //딕셔너리에 값이 없으면 2추가 (1이 아닌 2를 추가한 이유는 의상을 착용하지 않는 경우 때문)
            wearDic[wearKey] = 2
        }
    }
    if wearDic.values.count == 1{  //의상의 종류가 1개인 경우
        resultArray.append(wearDic.values.map{$0 - 1}[0])  //알몸의 경우 -1
    }
    else{  //의상의 종류가 1개 초과인 경우
        resultArray.append(wearDic.values.reduce(1){$0 * $1} - 1)  //알몸의 경우 -1
    }
    wearDic = [String:Int]()  //다음 testCase를 위해 빈 딕셔너리로 초기화
}

for i in resultArray{
    print(i)
}