영벨롭 개발 일지

[백준 BOJ][C++]15650번 N과 M(2) 풀이: 브루트 포스 & DFS 본문

알고리즘 문제 풀이/BOJ

[백준 BOJ][C++]15650번 N과 M(2) 풀이: 브루트 포스 & DFS

영벨롭 2022. 3. 28. 16:43

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

 

15650번: N과 M (2)

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

www.acmicpc.net

 

 이 문제는 15649번 N과 M(1) 풀이와 매우 유사합니다. 

(풀이: https://iridescent-zeal.tistory.com/86)

 

 하나만 바꾸어 주면 되는데요!

 

 dfs 함수에 현재 노드의 수를 나타내는 인자를 추가하여 해당 노드 보다 큰 수에 대해서만 자식 노드를 탐색하면 간단하게 풀립니다. 

 

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
#include<functional>
#include<cmath>

using namespace std;

int n, m;
bool visit[9] = { false, };
int arr[9];

void dfs(int num, int depth) {

	if (depth == m - 1) {
		for (int i = 0; i < m; i++) {
			printf("%d ", arr[i]);
		}
		printf("\n");

		return;
	}

	for (int i = num + 1; i <= n; i++) {
		if (!visit[i]) {
			visit[i] = true;
			arr[depth + 1] = i;
			dfs(i, depth + 1);
			visit[i] = false;
		}
	}

}

int main(void) {

	scanf("%d %d", &n, &m);

	for (int i = 1; i <= n; i++) {
		visit[i] = true;
		arr[0] = i;
		dfs(i, 0);
		visit[i] = false;
	}

	return 0;
}
반응형