[백준(python)] 1182번: 부분수열의 합

1182번: 부분수열의 합

문제

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

입력

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

출력

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

풀이 (브루트포스 알고리즘)

def dfs(depth, index):
    global count
    
    if 0 < depth <= N:
        if sum(result) == S:
            count += 1
        if depth == N:
            return
        
    for i in range(index, N):
        result.append(numList[i])
        dfs(depth+1, i+1)
        result.pop()

N, S = map(int, input().split())
numList = list(map(int, input().split()))
result = []
count = 0
dfs(0, 0)
print(count)

설명 (global 변수)

N과 S는 함수 내에서 그냥 사용했는데 count는 global변수로 설정한 이유

Python에서 함수 내에서 변수를 수정하려면 해당 변수를 전역 변수로 선언해야 한다. 그렇지 않으면 함수 내에서 새로운 지역 변수로 취급되어, 함수 외부의 동일한 이름을 가진 변수에 영향을 미치지 않음.

N과 S는 함수 내에서 수정되지 않기 때문에 함수 내에서 따로 전역 변수로 선언하지 않아도 잘 작동. 하지만 count 변수는 함수 내에서 값을 증가시키고 있기 때문에 함수 내에서 이 변수를 수정하려면 global count를 사용하여 전역 변수임을 명시해야 함.

따라서 count 변수는 함수 내에서 값이 변경되므로 함수 내에서 global count를 사용하여 전역 변수로 선언해주어야 함!