취준생의 코딩테스트 연습기

[SWEA] 4837 [파이썬 S/W 문제해결 기본] 2일차 - 부분집합의 합 / 파이썬(python) 본문

코딩테스트/SWEA

[SWEA] 4837 [파이썬 S/W 문제해결 기본] 2일차 - 부분집합의 합 / 파이썬(python)

Jiwon_C 2021. 3. 27. 21:24

※ SW Expert 아카데미의 문제를 무단 복제하는 것을 금지합니다.

# 문제 링크

swexpertacademy.com/main/learn/course/subjectDetail.do?courseId=AVuPDN86AAXw5UW6&subjectId=AWOVF-WqqecDFAWg&&

 

SW Expert Academy

SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!

swexpertacademy.com

# Soultion

1
2
3
4
5
6
7
8
9
10
11
12
import itertools
= int(input())
 
li = [i for i in range(1,13)]
for tc in range(1,t+1):
    n,k = map(int,input().split())
    combi = list(itertools.combinations(li,n))
    cnt = 0
    for i in combi:
        if sum(i)==k:
            cnt+=1
    print("#"+str(tc),cnt)
cs
 

1. 조합을 사용하기 위해 itertools 를 불러왔다.

4. 1부터 12까지 들어있는 리스트 생성

7. 조합을 이용하여 1~12가 들어있는 리스트에서 n개 뽑아내기

9. 각 숫자들의 합이 k이면 카운팅

Comments