[Python] 로또 - 백준

해당 글은 백준 6603번 문제 '로또'를 다룬다.

문제

https://www.acmicpc.net/problem/6603

6603번: 로또

 

설명

해당 문제는 itertools 라이브러리에 combinations 함수를 사용하면 쉽게 풀 수 있다.

한 줄씩 입력받고(line), 가장 첫번째 값은 k이므로 pop(0)을 통해 따로 빼두고, while문을 종료할 때만 사용하였다.

그리고 combinations 함수를 통해 S 중에 6개를 뽑는 경우의 모든 리스트를 뽑아두고, 이를 한 줄씩 출력해주면 된다.

 

 

코드

from itertools import combinations

while True:
    line = list(map(int, input().split()))
    k = line.pop(0)
    if k == 0:
        break
    combi = list(combinations(line, 6))
    for i in range(len(combi)):
        combi_new = [str(s) for s in combi[i]]
        print(' '.join(combi_new))
    print()