코딩테스트/SWEA
[SWEA] 4837 [파이썬 S/W 문제해결 기본] 2일차 - 부분집합의 합 / 파이썬(python)
Jiwon_C
2021. 3. 27. 21:24
※ SW Expert 아카데미의 문제를 무단 복제하는 것을 금지합니다.
# 문제 링크
SW Expert Academy
SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!
swexpertacademy.com
# Soultion
|
1
2
3
4
5
6
7
8
9
10
11
12
|
import itertools
t = 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이면 카운팅