[자료구조] 사탕 담기 파이썬

파이썬 조합을 활용해서 풀었는데

 

더 간편하게 푸는 방법이 있을 것 같다.

 

from itertools import combinations
def solution(m, weights):
    cnt = 0
    for i in range(1,len(weights)):
        COMB = combinations(weights,i)
        for comb in list(COMB):
            if sum(comb) == m:
                cnt += 1
    
    return cnt

 

조합을 구성하는 개수를 1부터 len(weights)까지 모든 조합을 순회하면서

 

조합내에서 만들어진 list의 원소들(tuple)의 합이 m인지 확인하면서 탐색한다.

 

효율적인 방법이 있을텐데