백준 15649_N과 M (1) 문제풀이 in C++
문제 URL
https://www.acmicpc.net/problem/15649
문제 요구사항
- 자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
- 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
- 첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
- 한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다.
- 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
- 수열은 사전 순으로 증가하는 순서로 출력해야 한다.
접근 방법
- DFS로 접근하며, 탐색 시 방문 체크를 하고, now +=1 을 하여 재 탐색한다. 탐색 종료 후, 백트래킹을 위해 방문 체크를 다시 false 한다.
풀이 순서
- N과 M을 입력 받는다.
- DFS함수에서 now(현재 수열 인덱스)를 매개변수로 다음을 수행한다.
- now값이 M(길이)와 같을 때, 수열의 내용을 출력한다.
- 그렇지 않다면, for문 i = 1부터, N까지 visited[ i ] 값이 false 라면, true로 변환 후, arr [ i ] 에 i 값을 넣고, 다음 탐색을 위해 DFS( now + 1)을 하여 탐색한다. 탐색이 종료되면, 계속해서 백트래킹을 하며 탐색 할 수 있도록 visited[ i ] 값을 false로 한다.
- 이와 같은 작업 반복
소스코드
#include <iostream>
using namespace std;
int N, M;
int arr[9];
bool visited[9];
void input() {
cin >> N >> M;
}
void solution(int now) {
if (now == M) {
for (int i = 0; i < M; i++)
cout << arr[i] << " ";
cout << "\n";
}
else {
for (int i = 1; i <= N; i++) {
if (!visited[i]) {
visited[i] = true;
arr[now] = i;
solution(now + 1);
visited[i] = false;
}
}
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
input();
solution(0);
return 0;
}