[백준알고리즘] 1182번: 부분수열의 합 -Python

728x90

1182번: 부분수열의 합

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

www.acmicpc.net

뻘짓했당

부분 수열이라고만 했는데 연속된 부분 수열인 줄 알고 계속 시도했었다.. 반례도 찾아가면서...

그러다가 아래의 반례를 찾고는.. 그냥 아무렇게나 뽑아서 만든 부분 수열이란 것을 알았다.

5 0
0 0 0 0 0

output:31

바로 combinations 메서드 써서 통과했다.

import sys

from itertools import combinations

n, s = map(int, sys.stdin.readline().split())

arr = list(map(int, sys.stdin.readline().split()))

cnt = 0

for r in range(1, n+1):

    cm = combinations(arr, r)

    for c in cm:

        if sum(c) == s:

            cnt += 1

print(cnt)

잘못된 점이나 부족한 점 지적해주시면 감사하겠습니다

728x90

공부노트