[백준/python] 6603번 로또

문제

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

6603번: 로또

 

코드

 from itertools import combinations

while True:
    no = list(map(int, input().split()))
    if no == [0]:
        break
    
    comb = list(combinations(no[1:], 6))
    for i in comb:
        print(*i)
    print('')

 

풀이

주어진 숫자 리스트에서 6개의 숫자를 고르는 조합 문제이다.

리스트를 받아올 때 첫 번째 숫자는 리스트의 길이를 나타내기 때문에 두번째 숫자부터 리스트로 받아와야한다는 점에 유의해야한다.

파이썬에서 조합을 위한 라이브러리와 함수를 지원하기 때문에 이를 활용하였다.

또한 제출 형식에 맞춰주기 위해 튜플 내의 원소를 풀어쓰기 위하여 *를 활용하였다.

0이 들어올 경우 반복문이 끝나야하므로 리스트가 '[0]' 리스트와 일치할 경우 break한다.