위장 - 프로그래머스 - 파이썬 풀이

문제 풀이

Counter을 통해 카테코리마다 몇개가 있는지 세어주고

카테고리 선택지 마다 안 입는 경우(1을 더해줌)를 포함해서 가능한 조합을 계산해주면 됩니다.

전부 안 입는 경우는 제외합니다.

직접 Counter에 해당하는 로직을 구현할 수 있지만, collections 에 있는 함수들은 애용해주는게 좋습니다!

클린 코드 작성법

1. 함수 같은 경우 동사형 get_something 으로 이름 붙여주면 가독성이 올라갑니다.
2. for 문 내에 i 같은 임시 변수 대신 cloth 같이 구체적으로 반복시킬 대상을 이름 붙여주면 좋습니다.
3. 주석을 통해 더 설명력 있는 문장을 전달할 수 있습니다. 코드만으로는 더 높은 추상화가 힘들 때 주석을 덧 붙여주면 좋습니다.
   1. choices -= 1
   2. 여기서 1은 전부 안 입는 경우인데 코드가 숫자가 들어가 있는 경우를 매직 넘버라고 합니다.
   3. 이 때 윗줄에 주석을 넣어 설명하거나 상수에 이름을 붙여 코드 자체로 설명할 수 있습니다.
      1. wearing_nothing_case = 1
      2. choices -= wearing_nothing_case

해답 코드

from collections import Counter

def get_category_counts(clothes):
    # 카테고리마다 몇개가 있는지 확인
    categories = []
    for cloth in clothes:
        category = cloth[1]
        categories.append(category)
    categories_counter = Counter(categories)
    category_counts = list(categories_counter.values())
    return category_counts

def get_possible_choices(category_counts):
    choices = 1
    for count in category_counts:
        choices *= (count+1)
    # 전부 안 입는 경우는 제외
    choices -= 1
    return choices

def solution(clothes):
    category_counts = get_category_counts(clothes)
    answer = get_possible_choices(category_counts)
    return answer

 

도움이 됐다면 공감버튼을 눌러주세요! 질문이 있다면 댓글 달아주세요!

 

문제 링크

알고리즘 연습 - 위장 | 프로그래머스