[백준/python] 6603번: 로또

6603번: 로또 (acmicpc.net)

6603번: 로또

단순히 입력받은 집합 중에서 6개만 조합하여 출력하면 되는 문제.

import sys
input = sys.stdin.readline
from itertools import combinations
while True:
    srr=list(map(int,input().split()))
    k=srr[0]
    if k==0: break
    ans=list(combinations(srr[1:],6))
    for i in range(len(ans)):
        print(' '.join(map(str,ans[i])))
    print(" ")

itertools 패키지의 combinations를 사용(조합)

테스트 케이스가 끝나는 조건도 적어줌

출력할 때 튜플째로 출력하는 것이 아닌 숫자 사이 띄우면서 출력해야하므로 join 사용

join사용할때 숫자면 안됨... map으로 str타입으로 변형