본문 바로가기
코딩/백준

알고리즘 - Python / 백준 - 6603번 : 로또

반응형

6603번: 로또 (acmicpc.net)

 

6603번: 로또

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

www.acmicpc.net

 


풀이

  • 입력 받은 수들을 배열에 저장한다
  • 이후 배열을 정렬하고 combinations 을 통해서 나올 수 있는 모든 경우를 구하고 출력한다.

 

import sys
from itertools import combinations

while True:
  n = list(map(int, sys.stdin.readline().split()))
  if n[0] == 0:
    break
  n.pop(0)
  n.sort()
  ans = list(combinations(n,6))
  for i in range(0,len(ans)):
    for t in range(0,5):
      print(ans[i][t],end=" ")
    print(ans[i][5],end="\n")
  print("",end="\n")

 

 

 

반응형