Coding Test/Programmers
위장 - 프로그래머스 - 파이썬 풀이
심재훈
2019. 7. 20. 17:00
문제 풀이
Counter을 통해 카테코리마다 몇개가 있는지 세어주고
카테고리 선택지 마다 안 입는 경우(1을 더해줌)를 포함해서 가능한 조합을 계산해주면 됩니다.
전부 안 입는 경우는 제외합니다.
직접 Counter에 해당하는 로직을 구현할 수 있지만, collections 에 있는 함수들은 애용해주는게 좋습니다!
클린 코드 작성법
- 함수 같은 경우 동사형 get_something 으로 이름 붙여주면 가독성이 올라갑니다.
- for 문 내에 i 같은 임시 변수 대신 cloth 같이 구체적으로 반복시킬 대상을 이름 붙여주면 좋습니다.
- 주석을 통해 더 설명력 있는 문장을 전달할 수 있습니다. 코드만으로는 더 높은 추상화가 힘들 때 주석을 덧 붙여주면 좋습니다.
- choices -= 1
- 여기서 1은 전부 안 입는 경우인데 코드가 숫자가 들어가 있는 경우를 매직 넘버라고 합니다.
- 이 때 윗줄에 주석을 넣어 설명하거나 상수에 이름을 붙여 코드 자체로 설명할 수 있습니다.
- wearing_nothing_case = 1
- 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
도움이 됐다면 공감버튼을 눌러주세요! 질문이 있다면 댓글 달아주세요!
알고리즘 연습 - 위장 | 프로그래머스
programmers.co.kr