6603번: 로또
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로
www.acmicpc.net
>>
itertools 라이브러리를 이용하여 조합 구하기
k와 s의 원소를 하나의 리스트에 모두 입력받고 첫번째 원소는 k 변수에 저장하고 나머지는 s집합 리스트에 저장한다.
s의 집합 원소들을 조합함수를 이용하여 6개 뽑을 수 있는 경우의 수를 모두 출력한다.
from itertools import combinations
while True:
a=list(map(int,input().split()))
k=a[0]
if k==0:
break
s=a[1:]
for i in combinations(s,6):
for j in i:
print(j,end=' ')
print('')
print('')
반응형
'Python > Beakjoon' 카테고리의 다른 글
2953번: 나는 요리사다 (0) | 2021.08.07 |
---|---|
1026: 보물 (0) | 2021.08.07 |
1010번: 다리놓기 (0) | 2021.08.06 |
3052번: 나머지 (0) | 2021.08.04 |
4153번: 직각삼각형 (0) | 2021.08.04 |