150414 n개의 원소중 i개를 고르는 모든 경우를 출력하는 코드

0번부터 차례대로 번호가 매겨진 n개의 원소중 3개를 골라 모든 경우를 출력하는 코드이다.

for i in xrange(n): for j in xrange(i+1, n): for k in xrange(j+1, n): print i,j,k  

이 코드는 문제가 있다. 일단 4개가 되면 4개의 for문을 써야하고 5개가 되면 5개의 for문을 써야한다

그럼 이코드를 재귀함수로 고친 코드를 살펴보자

def pick(n, picked, to_pick): if to_pick is 0: print picked return smallest = 0 if not len(picked) else picked[-1]+1 for i in xrange(smallest, n): picked.append(i) pick(n, picked, to_pick-1) picked.pop()

우리는 일단 앞에서 문제가 몇개씩 고르면 그만큼 for문도 증가하는 문제를 해결했다. 그리고 코드 가독성도 훨씬 좋아졌다.

그럼 오늘의 챌린지

근데 지금까지 코드들은 [1,2,3] [1,2,4] ... [1,2,9] 에서 다시 돌아갈때 [2, 3, 4]가 된다 [2, 1, 2] 로 모든 숫자를 세려면 어떻게 코드를 바꿔야할까? 그냥 smallest 없이 0으로 만들면 되네... 4자리부터는 진짜 오래걸린다 근데...