코딩 테스트/백준

[백준6603] 로또

땅어 2020. 3. 9. 15:17
728x90
300x250

문제

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

 

6603번: 로또

문제 독일 로또는 {1, 2, ..., 49}에서 수 6개를 고른다. 로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다. 예를 들어, k=8, S={1,2,3,5,8,13,21,34}인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2

www.acmicpc.net

 

생각의 흐름

  1. 주어진 수열에서 6개의 수를 고르는 것이므로 "조합"임을 파악
  2. itertools의 combinations 메소드를 이용해 6개로 이루어진 조합을 구한다.

 

깨달은 점

  1. itertools의 combinations 사용법 : combinations(list, pick할 갯수)
  2. ' '.join(열거형 변수) : ' '.join(map(str, case))

코드

from itertools import combinations

flag = True

while flag:
    line = list(map(int, input().split()))
    n = int(line[0])
    if n == 0:
        flag = False
        break

    for case in combinations(line[1:], 6):
        print(' '.join(map(str,case)))
    print('')

 

728x90
300x250