백준 (boj) 파이썬 - 1182 번 : 부분수열의 합

문제링크  :https://www.acmicpc.net/problem/1182

1182번: 부분수열의 합

 

 

 

itertools의 combination을 사용하여 풀었다. (자동으로 n개중에 r개를 뽑는 조합(combination)을 반환해주는... )

https://infinitt.tistory.com/114

순열과 조합 ( n! , nPr , nCr ) - (itertools) combinations, permutations + 중복순열, 중복조합

 

문제에서 N과 S를 입력으로 준다.

N개의 정수로 이루어진 수열중에, ?개를 뽑아 그 합이 S와 같아야 하므로 조합을 사용하면 된다.

 

from itertools import combinations
N, S = map(int,input().split())
arr = list(map(int,input().split()))
cnt = 0
for i in range(1,len(arr)+1):
    answer = list(combinations(arr,i))


    for idx in range(len(answer)):
        sum_partier = sum(answer[idx])
        if sum_partier == S :
            cnt += 1
    
print(cnt)