[python]백준 6603번: 로또

Problem

문제링크: www.acmicpc.net/problem/6603

6603번: 로또

알고리즘 분류:  수학, 조합론, 백트래킹, 재귀

---

Solution

이 문제는 하나의 집합(N) 속에서 6개의 숫자로 이루어진 모든 조합을 구하는 문제이다.

python 의 "itertools" 라이브러리에서 지원하는 *combinations 함수를 사용하면 쉽게 구할 수 있다.

---

Code

import itertools

while 1:
    N = list(map(int, input().split()))

    if N[0] == 0: break			# N에 0이 입력될때까지 반복
    del N[0]				# N에 처음 값(k)을 삭제

    for i in itertools.combinations(N, 6):	#combinations함수로 만들수 있는 조합을 만든 후
        print(' '.join(map(str, i)))	#문자열 형식으로 출력한다.
    print()		

 

*combinations 함수는 리스트의 조합의 결과를 반환한다.