프로그래머스 - 의상

박영빈·2023년 4월 30일

Programmers

목록 보기
4/43

의상


설명

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

from collections import Counter

def solution(clothes):
    c_type = []
    for t in clothes:
        c_type.append(t[1])
    c_dict = Counter(c_type)
    
    answer = 1
    for c in c_dict.keys():
        c_dict[c] += 1
    for i in c_dict.values():
        answer *= i
    answer = answer-1
    return answer
  • clothes의 모든 type을 리스트로 만든 뒤, 각각 개수를 센다.
  • 각 요소의 개수를 전부 곱한 뒤 1을 빼주면 모든 경우의 수를 알 수 있다.
profile
안녕하세요<br>반가워요<br>안녕히가세요

0개의 댓글