6603 - 로또
📌 문제
독일 로또는 에서 수 6개를 고른다.
로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다.
예를 들어, k=8, S=인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2,3,5,13,21], …, [3,5,8,13,21,34])
집합 S와 k가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.
📋 코드
from itertools import combinations
while True:
tmp = list(map(int, input().split()))
if tmp[0] == 0:
break
K, lotto = tmp[0], tmp[1:]
comb = list(combinations(lotto, 6))
comb.sort(key=lambda x: x)
[print(*c) for c in comb]
print()💡 한마디
combinations 라이브러리는 굉장히 강력한 것 같다.
