로또

백준 6603

• 독일 로또는 {1, 2, ..., 49}에서 수 6개를 고른다. 로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다.
• 집합 S와 k가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.
• 입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로 주어진다. 입력의 마지막 줄에는 0이 하나 주어진다.
• 각 테스트 케이스마다 수를 고르는 모든 방법을 출력한다. 이때, 사전 순으로 출력한다. 각 테스트 케이스 사이에는 빈 줄을 하나 출력한다.

입출력

접근 방식

: combinations 이용

알게된 점

재귀 및 백트래킹으로 풀 수 있다.

코드

def dfs(idx, cnt):
    if cnt == 6: # 6개를 다 선택했으면 출력
        print(*lotto)
        return # 이전으로 돌아가 

    for i in range(idx, len(nums)): # 해당 인덱스부터 마지막까지
        lotto[cnt] = nums[i] # 할당하고
        dfs(i+1, cnt+1) # 깊이있게 탐색

while True:
    nums = list(map(int, input().split()))
    if nums[0] == 0: 
        break
    k = nums[0]
    del nums[0]
    lotto = [0]*k
    dfs(0, 0) # 시작 인덱스와 카운트(깊이)
    print()

끝까지 갔다가 끝에서부터 돌아온다.