Python/Beakjoon
6603번 : 로또
hyunjoo
2021. 8. 7. 14:45
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('')
반응형