15650: N과 M(2)

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

15650번: N과 M (2)

설명

자연수 N과 M이 주어졌을 때, 다음 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램 작성. 

• 1부터 N까지 자연수 중에서 중복 없이 M개의 고른 수열
• 고른 수열은 오름 차순이어야 한다.

핵심 아이디어

• 백트래킹
• 15649번과 거의 동일한 문제이지만 이 문제에서는 [1,2], [2,1]과 같은 경우는 중복되는 것으로 보고 [1,2] 만 출력해야 한다. 

# 15649
n,m = list(map(int,input().split()))
arr = []
def dfs():
    if len(s)==m:
        print(' '.join(map(str,arr)))
        return
    
    for i in range(1, n+1):
        if i not in arr:
            arr.append(i)
            dfs()
            arr.pop()
dfs()

이전 코드와 달라진 점은 start 변수를 추가 주었다. 기존에는 1 ~ n까지 모든 숫자를 사용했지만, [2,1]와 같이 앞의 숫자가 뒤의 숫자보다 작은 경우 즉, [뒤, 앞] 를 제외하기 위해 start 부터 n까지 숫자를 사용했다. 그리고 재귀함수를 호출할 때는 i를 이용하여 자신의 다음 숫자를 부르게 된다. 

코드 구현

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
# 수열을 담을 배열
arr = []
def dfs(start):
	if len(arr) == m:
		print(' '.join(map(str, arr)))
		return # 리턴하면 호출 함수로 돌아가 다음 코드 수행
	else:
		for i in range(start, n+1):
			if i not in arr:
				arr.append(i)
				dfs(i+1)
				arr.pop()
dfs(1)

출처: https://jiwon-coding.tistory.com/22