문제:
N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.
입력:
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1<=N<=20, |S|<=1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
출력:
첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.
풀이방법:
알고리즘 분류가 브루트 포스로 되어 있었기 때문에 가능한 모든 경우의 수를 만들고자 하였다. 따라서 itertools의 combinations을 사용하였다. itertools.combinations( iterable, n)과 같이 사용하며 iterable한 객체로 반환을 한다. 따라서 이를 반복문으로 넣고 합했을 때 s가 되는 경우에 answer를 하나씩 올렸다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
import itertools
n,s=map(int,input().split())
numbers=list(map(int,input().split()))
answer=0
for i in range(1,n+1):
for c in itertools.combinations(numbers,i):
if s == sum(c):
answer+=1
print(answer)
|
cs |
문제링크:
https://www.acmicpc.net/problem/1182
1182번: 부분수열의 합
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
www.acmicpc.net
'Algorithm > Python' 카테고리의 다른 글
| [BOJ]13305. 주유소 (0) | 2020.02.25 |
|---|---|
| [BOJ]2485. 가로수 (0) | 2020.02.20 |
| [BOJ]1431. 시리얼 번호 (1) | 2020.02.13 |
| [BOJ]6603. 로또 (0) | 2020.02.11 |
| [BOJ]1297. TV 크기 (0) | 2020.02.06 |