(프로그래머스) 코딩테스트 연습 - 해시 - 위장 (Python)

 

문제 : https://programmers.co.kr/learn/courses/30/lessons/42578

코딩테스트 연습 - 위장 | 프로그래머스

문제설명

 

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

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

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

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

 

제한사항

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

제가 통과한 코드

from collections import Counter

def solution(clothes):
    answer = 1
    kind = Counter([cloth[1] for cloth in clothes])
    
    for i in kind:
           answer *= (kind[i] + 1)
           
    answer -= 1
    
    return answer

코딩테스트를 전혀 수학적으로 생각하지 않았는데,, 수학적으로 푸는 거더라구요...

어떻게 접근해서 풀어야할지 몰라서 검색하니 결국 경우의 수를 이용해서 푸는 거 일줄이야..

학창시절에도 경우의수, 순열, 조합 이 부분이 통계 확률 과목에서 제일 약했던 부분이라 그런지 어떻게 식을 세워야 할지 모르겠더라구요 ㅠㅠ 

 

풀이

옷의 종류가 A, B, 이고 각 2개씩 가지고 있다면, 2*2 = 4가지의 경우의 수

이것은 [A1, B1], [A1, B2], [A2, B1], [A2, B2]의 경우의 수입니다.

그렇다면, 스파이는 최소 1개의 옷을 입어야 하니 각각 갯수를 1개씩 더 더해주면 3*3 = 9로 9가지의 경우의 수가 나옵니다.

 

이 경우에는 [없음, 없음], [A1, 없음], [A2, 없음] .... 이런식으로 되기 때문에 스파이가 아예 입지 않는 경우인 1을 빼줍니다.

 

이렇게 수식을 세워서 알고리즘을 짜면 될 것 같습니다.

 

항상 경우의 수는 풀이를 보면 이해하지만, 보지 않으면 잘 세워지지 않더라구요... 열심히 공부해야겠습니다.