[Brute Force][DFS][Backtracking] 백준 1182번 : 부분수열의 합

[jeongabae]

실버2

조합함수 이용해서 푸는게 제일 빠름!

시간 제한 | 메모리 제한 | 제출 | 정답 | 맞힌 사람 | 정답 비율 -- | -- | -- | -- | -- | -- 2 초 | 256 MB | 54997 | 25324 | 16358 | 44.284%

문제

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

입력

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

출력

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

예제 입력 1 복사

5 0
-7 -3 -2 5 8

예제 출력 1 복사

1

출처

• 문제를 만든 사람: author5
• 문제의 오타를 찾은 사람: eric00513, jh05013
• 데이터를 추가한 사람: rdd6584
• 잘못된 데이터를 찾은 사람: tncks0121

알고리즘 분류

• 브루트포스 알고리즘
• 백트래킹

내 풀이 : 시간 오래 걸리지만 통과는 함...

아래 다른 사람 풀이처럼 푸는게 제일 좋을 듯! N과 M문제 푼거랑 비슷하게 푸니까 오래 걸렸다.. 처음부터 조합함수 이용 풀이가 생각나긴했지만 공부 겸 이걸로 풀자해서 풀었는데 ㅠ..

import sys
def input():
    return map(int, sys.stdin.readline().split())
N, S = input()
a = list(input())
used = [0] * (N)
ans = []
cnt = 0

def dfs(idx,start):
    global cnt
    if idx>N:
        return
    if len(ans) and sum(ans)==S:
        cnt +=1

    for i in range(start, N):
        if not used[i]:
            used[i] = 1
            ans.append(a[i])
            dfs(idx + 1,i+1)
            used[i] = 0
            ans.pop()
dfs(0,0)
print(cnt)

다른 사람 풀이 1

재귀함수로 구현을 해준다. dfs(idx, sum)을 했을때, sum에 해당 인덱스를 포함하는 dfs와 포함하지 않는 dfs를 실행시킨다.

메모리 30840 KB 시간 548ms

import sys
input = sys.stdin.readline
def dfs(idx, sum):
    global cnt
    if idx >= n:
        return
    sum += s_[idx]
    if sum == s:
        cnt += 1
    dfs(idx + 1, sum - s_[idx])
    dfs(idx + 1, sum)
n, s = map(int, input().split())
s_ = list(map(int, input().split()))
cnt = 0
dfs(0, 0)
print(cnt)

다른 풀이 : 더 빠른 풀이는 당연히 조합 함수 이용

메모리 30840 KB 시간 416ms

import sys
from itertools import combinations

input = sys.stdin.readline
n, s = map(int, input().split())
arr = list(map(int, input().split()))
cnt = 0
for i in range(1, n+1):
    comb = combinations(arr, i)

    for x in comb:
        if sum(x) == s:
            cnt += 1

print(cnt)