Algorithm (PS)

[백준] 6603 로또 Python

minjiwoo 2022. 10. 9. 17:33
728x90

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

 

6603번: 로또

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

www.acmicpc.net

쉬어가는 경우의 수 문제~ 

조합 모듈 combinations를 이용하면 간단하게 풀 수 있다. 

from itertools import combinations

while True:
    data = list(map(int, input().split()))
    if len(data) == 1:
        break
    array = list(combinations(data[1:], 6))
    array.sort()
    for case in array:
        print(*case)
    print()
728x90