[백준] 1182 : 부분수열의 합 - Python

부분수열의 합

N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.

입력

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

출력

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

import sys
sys.stdin = open("input.text", "rt")
from collections import deque

n,s = map(int, input().split())
data = list(map(int, input().split()))
cnt = 0
res = []
def dfs(L, res):
    global cnt
    if L == n: # 모두 확인. 종료조건
        if sum(res) == s and len(res) != 0:
            cnt += 1
    else:
        dfs(L+1, res + [data[L]])
        dfs(L+1, res)
dfs(0,res)
print(cnt)

코멘트

처음에는 dfs(L,sum) 이런 식으로 합으로 계산했는데 첫 예시에 합이 0이되는 경우에는 아무것도 선택을 안할 때도 가능했다. 그렇기에 체크하는 변수를 하나 추가해서 확인하거나 했어야 했는데 이렇게 리스트를 넣고 다음으로 넘어갈 때 위와 같이 추가해주는 방식을 사용해도 된다.

• 그리고 모두 확인한 경우에 개수를 통해 0이 아니라면 검사를 진행하면 된다.

m개를 선택해야 하는 조합 류의 문제는 종료조건이 m개를 선택하면 끝나는게 맞지만, 이런 문제처럼 속한다, 속하지 않는다 류의 문제는 모든 리스트의 데이터를 다 확인하러 가야한다. 즉 최종 마지막 인덱스까지 접근하는 것이 종료조건이다. 문제를 잘 확인하면서 접근하자.