[BOJ] 15650 N과 M(2)

date

Jul 5, 2023

slug

boj-15650

author

status

Public

문제 링크

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

문제

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

1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열

고른 수열은 오름차순이어야 한다.

문제풀이

DFS를 통해

탐색하는 깊이가 2일경우 출력하도록 한다.

이때 탐색하는 노드가 부모노드보다 클경우 탐색하지 않는다.

파이썬 문제풀이 코드


def dfs(st, arr, depth, limit):
    if depth == limit:
        for i in st:
            print(i, end=" ")
        print()
        return
    
    for i in arr:
        if st[-1] < i:
            st.append(i)
            dfs(st, arr, depth+1, limit)
            st.pop(-1)
    
    
    
M, N = tuple(map(int, input().split()))

arr = range(M+1)[1:] # 1부터 오름차순

for i in arr:
    st = []
    st.append(i)
    dfs(st, arr, 1, N)

시간복잡도

재귀탐색

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

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

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

예제 입력

4 2

예제 출력

알고리즘 분류

백트래킹