[코딩테스트] 백준 부분수열의 합 파이썬(Python)

https://www.acmicpc.net/problem/1182

1182번: 부분수열의 합

from itertools import combinations
n, s = map(int, input().split())
array = list(map(int, input().split()))
result = 0
for i in range(1, n+1):
  for numbers in combinations(array, i):
    if sum(numbers) == s:
      result += 1
print(result)

 

반복문으로 부분수열의 크기를 1부터 n까지

combinations 모듈을 사용하여 모든 부분수열(numbers)을 구하였습니다.

구해진 부분수열(numbers)의 합이 주어진 입력 s와 동일하다면 result + 1을 하여 개수를 증가시켰습니다.

마지막으로 print(result)를 하여 문제를 해결하였습니다.