사탕담기

문제

그램(gram)을 담을 수 있는 가방에 사탕을 가득 채우는 경우의 수를 구하는 문제

코드

def solution(m, weights):
    answer = 0

    for i in range(len(weights)):
        answer += list(map(sum, combinations(weights, i))).count(m)
    
    return answer

새로 알게된 지식

for i in range(len(weights)):
	answer += list(map(sum, combinations(weights, i))).count(m)

map()을 이용하여 weights의 원소들의 경우의 수들을 합한 값들을 리스트로 생성
answer에는 m값이 있는 만큼 카운트하여 더해서 넣음

출처

https://school.programmers.co.kr/courses/13093/lessons/88779

좋은 웹페이지 즐겨찾기