@@@ 알고리즘/백준 스터디

1182(부분 수열의 합) - 해결

HTG 2021. 8. 12. 15:50
728x90

부분수열의 합

 

문제

N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 

그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 

둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

 

출력

첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.


SSAFY에서 배운 방법으로 하니 바로 풀렸다 근데 브루트포스는 맞는데 백트랙킹은 된건지 모르겠다...

import sys

N, S = map(int,sys.stdin.readline().split())

nums = list(map(int,sys.stdin.readline().split()))
cnt = 0

for i in range(1,1<<N):
    nsum = 0
    for j in range(N):
        if i & (1<<j):
            nsum += nums[j]
    if nsum == S:
        cnt += 1

print(cnt)