[백준 15666번] N과 M(12) C++ 풀이

시간 제한

1초

메모리 제한

512MB

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
N개의 자연수 중에서 M개를 고른 수열
같은 수를 여러 번 골라도 된다.
고른 수열은 비내림차순이어야 한다.
(길이가 K인 수열 A가 A1 ≤ A2 ≤ … ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.)

입력

첫째 줄에 N과 M이 주어진다.(1 ≤ M ≤ N ≤ 8)
둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

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

내 맞은 풀이

#include<bits/stdc++.h>
using namespace std;

int n(0), m(0);
int inputSeq[10], outputSeq[10];

void makeSeq(int cur, int latestIdx) {
	if (cur == m) {
		for (int i = 0; i < m; i++)
			cout << outputSeq[i] << ' ';
		cout << '\n';
		return;
	}
	int latestNum(-1);
	for (int i = latestIdx; i < n; i++) {
		if (latestNum == inputSeq[i])
			continue;
		outputSeq[cur] = inputSeq[i];
		latestNum = inputSeq[i];
		makeSeq(cur + 1, i);
	}
}
int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);

	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> inputSeq[i];
	sort(inputSeq, inputSeq + n);
	makeSeq(0, 0);
	return 0;
}

백트래킹을 이용한다.
input에 중복되는 숫자가 들어올 수 있고, 하나의 숫자를 여러 번 사용할 수 있으므로 중복되는 순열을 구하는 것을 피해야 한다.
이를 위해 어떤 루프에서 사용한 숫자를 다음 루프에서 다시 사용하는지를 점검하면 되므로 int latestNum을 함수 내부에서 선언하여 이용한다.
또 비내림차순이 조건이므로 마지막 index를 다음 함수에 전달해 다음 for문의 시작점을 설정한다.