[알고리즘/Swift] 프로그래머스 - 위장 (해시)

Python3 로 풀었던 문제를 Swift 로 다시 풀어봤어요. iOS 개발을 좀더 효율적으로 하기위해서 Swift 데이터구조, 알고리즘 공부의 필요성을 느꼈습니다. 🤔 '위장' 문제에 대한 접근 방법은 이전 포스팅을 참고해주세요~

[알고리즘/Python] 프로그래머스 - 위장 (해시)

✔︎ Dictionary 를 이용한 풀이 방법

스파이가 가진 의상들이 담긴 2차원 배열 clothes 은 [의상의 이름, 의상의 종류] 형태로 되어있습니다.

여기서 의상의 이름은 중복되지 않고, 의상의 종류는 중복될 수 있습니다. 👀

 

먼저, [의상의 종류(String): 의상의 개수(Int)] 형태로 딕셔너리를 초기화해줬어요! 

• 의상의 이름은 중복되지 않으므로 개수만 세면 됩니다.
• 의상의 종류는 중복될 수 있으므로 key 가 존재하지 않으면 0으로 초기화를 하고, 존재하면 1을 더했습니다. 

의상의 종류에 해당하는 개수를 구한 뒤에는, 이 값들을 통해 전체 조합의 개수를 구했습니다.

• 의상의 개수가 A, B, C 라면 -> 전체 조합의 개수는 (A+1) * (B+1) * (C+1) - 1 입니다.
• 하나씩 입는 경우도 포함하기 위해서 num + 1 의 값을 곱했습니다. 
• 1 을 빼주는 이유는, 공집합(옷을 안입는경우(?))을 제외하기 위함입니다.

전체 코드는 다음과 같습니다.

import Foundation

func solution(_ clothes:[[String]]) -> Int {

    // 딕셔너리 초기화
    var clothesDict : [String: Int] = [:]
    for cloth in clothes {
        let key = cloth[1]
        if !clothesDict.keys.contains(key) {
            clothesDict[key] = 0
        }
        clothesDict[key]? += 1
    }

    // 서로 다른 옷의 조합의 개수 구하기
    var count = 1
    for num in clothesDict.values {
        count *= (num + 1)
    }
    return count-1
}

 

✔︎ Map, Filter, Reduce 를 이용한 풀이 방법

다른 풀이도 학습하고 싶어서 프로그래머스 풀이의 상단에 있는 풀이를 참고했습니다.

먼저, compactMap 함수를 통해서clothes 배열에서 의상의 종류만 따로 배열로 만들어줍니다.

Set 자료형으로 만든 이유는 의상의 종류의 중복을 제거하기 위합입니다.

💡 map, compactMap
map 은 주어진 값을 하나씩 매핑하는데 이용하는 함수입니다. 
compactMap 은 map 과 같은 역할을 하지만 nil 을 제거하고 옵셔널 바인딩을 해주는 함수입니다. 

이후, 의상 종류에 해당하는 개수를 세기 위해서 map, filter 함수를 이용합니다.

counts 배열의 각 원소의 값은 의상 종류의 개수에 1을 더한 값입니다.

 

마지막으로, 전체 의상의 조합 개수를 구하기 위해서 reduce 함수를 이용합니다.

counts 배열을 반복하며 각 원소들을 곱해주고, 1을 빼주는 과정과 같습니다.

---

• map, flatMap, compactMap 에 대한 블로그
• 프로그래머스 문제