본문 바로가기
문제 풀이/백준(BOJ)

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

by 희조당 2022. 7. 31.
728x90

https://www.acmicpc.net/problem/1182

 

1182번: 부분수열의 합

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

www.acmicpc.net


💡 문제 풀이

브루트 포스 문제이다.

 

부분 수열을 만들어서 합이 목표값과 같은지만 확인해주면 된다.

itertools의 combination을 사용하면 부분 수열을 쉽게 만들 수 있다.

✔️ 느낀 점

그렇게 어려운 문제는 아니었다.

💻 코드

import sys
from itertools import combinations
input = sys.stdin.readline

N, M = map(int, input().split())
arr = list(map(int, input().split()))
cnt = 0

for i in range(1,N+1):
    c = combinations(arr, i)
    
    for n in c:
        if sum(n) == M: cnt+=1

print(cnt)

댓글