STUDY/Algorithm

[백준] 1182 부분 수열의 합, python

sinawi95 2022. 1. 3. 12:43
728x90

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

 

1182번: 부분수열의 합

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

www.acmicpc.net

비트마스크를 사용한 조합문제이다.

dp를 사용하지 않았고 비트마스크를 사용해서 모든 부분 수열의 합을 구했다.

def main():
    N, S = map(int, input().split())
    seq = list(map(int, input().split()))

    answer = 0
    for select in range(1, 1 << N):
        tmp = 0
        for i in range(N):
            if select & (1 << i):
                tmp += seq[i]
        if tmp == S:
            answer += 1

    print(answer)

if __name__ == '__main__':
    main()

'STUDY > Algorithm' 카테고리의 다른 글

[백준] 17404 RGB거리 2, python  (0) 2022.01.04
[백준] 5430 AC, C++  (0) 2022.01.03
[백준] 2098 외판원 순회, python  (0) 2022.01.03
[백준] 1311 할 일 정하기 1, python, C++  (0) 2022.01.02
[백준] 9251 LCS, python, C++  (0) 2022.01.02