티끌모아 태산

15650: N과 M(2) 본문

백준 문제/백트래킹

15650: N과 M(2)

goldpig 2024. 3. 11. 14:56
728x90

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

 

15650번: N과 M (2)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

설명

자연수 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

 

728x90

'백준 문제 > 백트래킹' 카테고리의 다른 글

15652: N과 M(4)  (0) 2024.03.11
15651: N과 M(3)  (0) 2024.03.11
15649: N과 M (1)  (0) 2024.03.11