프로그래밍
PS
[백준 / python] 1182: 부분수열의 합
2024년 05월 14일 by Hwang
https://www.acmicpc.net/problem/1182
문제
N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
출력
첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.
풀이
N이 최대 20이므로 모든 조합의 개수를 구하는데 많은 연산이 필요하지 않기 때문에 combinations을 이용해서 모든 조합을 구해서 풀었습니다
import sys
from itertools import combinations
N, S = map(int, sys.stdin.readline().split())
arr = list(map(int, sys.stdin.readline().strip().split()))
result = 0
for i in range(1, N + 1):
for com in combinations(arr, i):
current_sum = sum(com)
if current_sum == S:
result += 1
print(result)
Comment