알고리즘💻/Bruteforce&Backtracking
BOJ 1182번: 부분수열의 합
호프
2021. 7. 25. 20:45
https://www.acmicpc.net/problem/1182
1182번: 부분수열의 합
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
www.acmicpc.net
이전에 풀었던 문제인데, 백트레킹 문제를 너무 오랜만에 풀어서 감이 잘 오지 않았다. 이제는 주먹구구식으로 풀기보다 유형별로 정리해서 공부를 제대로 해야겠다..
재귀를 이용해서 풀 수 있었다.
import sys
sys.setrecursionlimit(10**8)
input = sys.stdin.readline
N, S = map(int, input().split())
num = list(map(int, input().split()))
cnt = 0
def solve(i, total):
global cnt
if (i == N):
if (total == S):
cnt += 1
return
solve(i+1, total)
solve(i+1, total+num[i])
solve(0, 0)
if (S==0):
cnt -= 1
print(cnt)