해리의 데브로그

(SW 문제해결 기본 - LIST2) SWEA 4837 - 부분집합의 합

05 Mar 2019 | Algorithm SWEA

문제

  • SWEA 4837 - [파이썬 S/W 문제해결 기본 2일차] LIST2 - 부분집합의 합
  • 문제링크
  • 문제의 저작권은 SW Expert Academy에 있습니다.

나의 코드

  • 부분집합을 구할 때, 비트연산자를 이용하였음.
TC  = int(input())
num = [1,2,3,4,5,6,7,8,9,10,11,12]
Len = len(num)

#부분집합 구하기
lst = []
for i in range(1<<Len):
    sub_lst = []
    for j in range(Len):
        if i & (1<<j):
            sub_lst.append(num[j])
    lst.append(sub_lst)


for tc in range(1, TC+1):
    N, K = map(int, input().split())

    #길이 맞는 리스트 구하기
    len_lst = []
    for i in lst:
        if len(i) == N:
            len_lst.append(i)


    #합 일치 유무 확인
    result = 0
    for i in len_lst:
        if sum(i) == K:
            result += 1

    print('#%s %d'%(tc, result))

Related Posts

Comments