백준: 6603 로또

문제

6603번: 로또

 

문제 풀이

 집합이 주어지면 로또의 경우 6개의 수를 뽑기 때문에 6개의 수를 선택할 수 있는 경우의 수를 선택하는 문제이다. 예제 출력을 보면, 6개를 선택할 수 있는 조합을 찾는다는 것을 알 수 있다. 앞서 다룬 N과 M 시리즈와 동일한 방식으로 조합을 구하면 된다.

 

코드

DFS를 사용한 문제 풀이

from sys import stdin


def dfs(idx, depth):
    global answer

    if depth == 6:
        answer.append([nums[i] for i in check])
    else:
        for i in range(idx, n):
            if i in check:
                continue
            check[depth] = i
            dfs(i + 1, depth + 1)
            check[depth] = -1

if __name__ == '__main__':
    while True:
        answer = []
        nums = list(map(int, stdin.readline().split()))
        if not nums[0]:
            break

        n = nums[0]
        nums = nums[1:]
        check = [-1] * 6
        dfs(0, 0)
        for num in answer:
            print(*num)
        print()

 

itertools를 활용한 문제 풀이

from itertools import combinations
while True:
    k, *nums = map(int, input().split())
    if not k:
        break
    for cases in combinations(nums, 6):
        for case in cases:
            print(case, end=' ')
        print()
    print()