프로그래머스 / 해시 / 위장 / level 2 / 파이썬

programmers.co.kr/learn/courses/30/lessons/42578

코딩테스트 연습 - 위장

 

 

문제 설명

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

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

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

 

 

제한사항

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

 

 

입출력 예

 

 

입출력 설명

예제1

예제2

 

 

나의 문제풀이 과정

• 모든 조합의 수를 구하는 문제입니다.
• (옷 종류별 옷가지 개수 + 1)를 모두 곱해줍니다.
• (해당 종류의 옷을 안입는 경우도 포함하기 때문에 + 1)
• 그 어느것도 입지 않은 상태까지도 계산되었으니 마지막에 -1 한 값을 return합니다.

 

 

나의 풀이 코드

from collections import defaultdict
from functools import reduce

def solution(clothes):
    closet = defaultdict(int)
    for name, ctg in clothes:
        closet[ctg] += 1
    values = [x+1 for x in closet.values()]
    return reduce(lambda x,y: x*y, values) - 1

 

 

결과

 

 

Counter 클래스 사용하기 (collections 모듈)

Counter를 활용하면 앞선 풀이에서 직접 구현한 딕셔너리 기능을 좀 더 간단하게 구현할 수 있습니다.

Counter 설명 보기

[파이썬] collections 모듈의 Counter 클래스 사용법

from collections import Counter
from functools import reduce

def solution(clothes):
    closet = Counter([ctg for name, ctg in clothes])
    values = [x+1 for x in closet.values()]
    return reduce(lambda x,y: x*y, values) - 1

 

Counter 사용 결과

딕셔너리를 직접 구현한 것보다 시간이 조금 더 걸리는 게 보이네요.

그래도 이 문제는 input의 제한폭이 넓지 않아서 이렇게 풀어도 크게 문제는 없어 보입니다.

 

 

mul 함수 사용하기 (operator 모듈)

또, 앞서 사용한 람다식을 mul로 대체할 수도 있습니다.

(operator.mul(a, b)는 a * b를 반환합니다.)

from functools import reduce
from operator import mul
from collections import defaultdict

def solution(clothes):
    closet = defaultdict(int)
    for name, ctg in clothes:
        closet[ctg] += 1
    values = [x+1 for x in closet.values()]
    return reduce(mul, values) - 1