Adventure Time - Finn 3
본문 바로가기
코테연습/python

합이 같은 부분집합(DFS : 아마존 인터뷰)

by hyun9_9 2026. 5. 13.

 N개의 원소로 구성된 자연수 집합이 주어지면, 이 집합을 두 개의 부분집합으로 나누었을 때 두 부분집합의 원소의 합이 서로 같은 경우가 존재하면 “YES"를 출력하고, 그렇지 않으면 ”NO"를 출력하는 프로그램을 작성하세요. 둘로 나뉘는 두 부분집합은 서로소 집합이며, 두 부분집합을 합하면 입력으로 주어진 원래의 집합이 되어 합니다. 예를 들어 {1, 3, 5, 6, 7, 10}이 입력되면 {1, 3, 5, 7} = {6, 10} 으로 두 부분집합의 합이 16으로 같은 경우가 존재하는 것을 알 수 있다.

import sys
# sys.stdin=open("input.txt","rt")
# n,m = map(int,input().split())
# n = int(input())
# arr = list(map(int,input().split()))

def DFS(L,sum):
    if sum>total//2:
        return
    if L ==n:
        if sum == (total-sum):
            print("YES")
            sys.exit(0)
    else:
        DFS(L+1,sum+arr[L])
        DFS(L+1,sum)




if __name__ =="__main__":
    n = int(input())
    arr = list(map(int,input().split()))
    total=sum(arr)
    DFS(0,0)
    print("NO")

'코테연습 > python' 카테고리의 다른 글

동전교환  (0) 2026.05.16
중복순열 구하기  (0) 2026.05.14
부분집합 구하기(DFS)  (0) 2026.05.12
이진트리 순회(깊이우선탐색)  (0) 2026.05.11
재귀함수를 이용한 이진수 출력  (0) 2026.05.09