본문 바로가기
백준 알고리즘

파이썬) 백준 알고리즘 | 6603번 : 로또

by 코딩새내기_ 2022. 3. 3.

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

 

6603번: 로또

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로

www.acmicpc.net

백준 6603 '로또' 문제입니다.

itertools 모듈의 combination을 이용하면 간단하게 구현할 수 있습니다.

import sys
from itertools import combinations
input = sys.stdin.readline
while True:
    x = list(map(int, input().split()))
    if x[0] == 0:
        break
    num = x[0]
    nums = x[1:]
    result = list(combinations(nums,6))
    for re in result:
        print(*re)
    print()

댓글