해시 - 위장

아래 문제는 프로그래머스에서 제공하는 코딩테스트 > 해시 > 위장 파트의 문제입니다🧑🏻‍💻

문제 제시

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

종류이름

얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항

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

문제 해결

import Foundation

func solution(_ clothes:[[String]]) -> Int {
    let types = Set(clothes.map{ $0.last })
    
    let counts = types.map{ types in
        return clothes.filter{ $0.last == types }.count + 1
    }
    return counts.reduce(1) { $0 * $1 } - 1
}

사용된 개념

 - 고차함수 (map, filter, reduce)

문제 뒷담화

문제를 접하고 우선 필요한것은 아이템의 종류의 수와 그 종류에 속한 아이템의 각 개수 파악이였다.

그러기위해 우선 받아온 2차원 배열의 아이템의 종류 파악을 위해 map을 사용하였다. map을 사용하면 해당 데이터를 다시 재정의하여 매핑시켜줄 수 있다. 지금 생각해보니 map도 되지만 2차원 배열이니 compactMap을 사용하는게 더 괜찮았을듯 싶다. 그렇게 매핑을 해주고 중복된 종류에 대해 중복을 없애주기 위해 Set 즉 집합을 이용하였다. array, set, dictionary의 집단자료형중 순서가 필요없고 중복된 교집합의 값을 없애주는 기능을 하는 set이 알맞았다. 

이렇게 완료된 아이템 종류에 대한 자료를 types로 담아주었다.

그 뒤, 전체 아이템의 타입을 방금 만들어준 타입과 비교하여 어느종류에 몇개가 들어있는지 반환해주기 위해 우선 types를 map을 사용해전체 돌 수 있도록 해주고 return으로 clothes 2차원 배열을 타입이 같은지 파악하고 같으면 넣어주는 filter를 사용했다.

그렇게 되면 필터된 clothes 배열의 아이템 수를 파악하기 위해 count를 사용해주고 하나를 더 더해 counts 배열에 최종적으로 담아주었다.

+1을 해주는 이유는 해당 종류가 2가지고 각 종류의 아이템 수가 3개씩이라면 한종류만 선택할 경우 두종류 다 선택할 경우가 있기에 안입는 경우도 포함하여 주어야 하기 때문에 +1을 해준것이다.

그렇게 된다면 위장 의상의 종류에서 한종류는 안입고 한종류는 입고 하는 경우도 따로 계산해줄 필요가 없다. 즉 동시에 입는다 치는셈이다. (안입는 경우도 nil값 처럼 동시에 진행된다고 보는것이다.)

그렇게되면 각 종류마다 아이템 갯수를 곱해주면 전체 경우의 수가 나온다.

마지막으로, 경우의 수 도출값을 구하기 위해 reduce를 사용하여 초기값은 1로주고 counts에 포함된 각 아이템의 수를 더해준다.

아참, 그리고 모두 안입는 경우는 없다고 했으니 -1을 해줘서 답을 도출해내면된다.

 

 

[참고자료]

프로그래머스 코딩테스트 연습 > 해시 > 위장 파트

programmers.co.kr/learn/courses/30/lessons/42578?language=swift#