거북이의 IT 공부

[백준 15652] N과 M(4) - C++ / 백트래킹, 재귀 알고리즘 본문

Baekjoon

[백준 15652] N과 M(4) - C++ / 백트래킹, 재귀 알고리즘

버니빈 2020. 5. 23. 22:02

문제

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

 

15652번: N과 M (4)

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

www.acmicpc.net

나의 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#include <iostream>
#include <vector>
#define MAX 8
 
using namespace std;
 
int arr[MAX];  //수열 숫자
vector<int> vec;  //수열
bool visited[MAX][MAX];  //각 cnt단계의 방문여부 - 0으로 초기화됨
int cnt, n, m;
 
void Print() {
    for (int i = 0; i < vec.size(); i++) {
        cout << vec[i] << ' ';
    }
    cout << '\n';
}
 
void DFS(int cnt, int s) {
    if (cnt == m) {
        Print();
        return;
    }
 
    for (int i = s; i < n; i++) {
        if (visited[cnt][i]) continue;
        visited[cnt][i] = true;
        vec.push_back(arr[i]);
        DFS(cnt + 1, arr[i] - 1);
        visited[cnt][i] = false;
        vec.pop_back();
    }
}
 
int main(void) {
    cin >> n >> m;
 
    for (int i = 0; i < n; i++) {
        arr[i] = i + 1;
    }
 
    DFS(00);
}
cs

 

여기서 핵심은 int s이다.

그 이전 단계의 숫자부터 탐색을 시작하면 되므로

예를 들어)

1 -> 1 -> 1 

2 -> 2 -> 2 ( cnt = 1인 단계에서 그 이전 단계 숫자부터 시작해서 2이고, cnt = 2인 단계에서 그 이전 단계 숫자가 2이므로 2부터 시작해서 탐색한다!!)

Comments