[백준 15665번] N과 M(11) C++ 풀이

시간 제한

1초

메모리 제한

512MB

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
N개의 자연수 중에서 M개를 고른 수열
같은 수를 여러 번 골라도 된다.

입력

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

출력

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

내 맞은 풀이

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

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

void makeSeq(int cur) {
	if (cur == m) {
		for (int i = 0; i < m; i++)
			cout << outputSeq[i] << ' ';
		cout << '\n';
		return; 
	}
	int latestNum(-1);
	for (int i = 0; i < n; i++) {
		if (latestNum != inputSeq[i]) {
			outputSeq[cur] = inputSeq[i];
			latestNum = inputSeq[i];
			makeSeq(cur + 1);
		}
	}
}

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);
	return 0;
}

백트래킹을 이용한다.
input에 중복되는 숫자가 들어올 수 있고, 하나의 숫자를 여러 번 사용할 수 있으므로 중복되는 순열을 구하는 것만 피하면 된다.
이를 위해 어떤 루프에서 사용한 숫자를 다음 루프에서 다시 사용하는지를 점검하면 되므로 int latestNum을 함수 내부에서 선언하여 이용한다.