백준 15665 : N과 M(11)

문제 링크입니다 https://www.acmicpc.net/problem/15665

15665번: N과 M (11)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 문제였습니다. 주어지는 N개의 자연수 중 같은 수가 있을 수도 있습니다. 조건은 다음과 같고,

 

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.

'15664번 N과 M(10)' 문제(https://seokeeee.tistory.com/152)에서 중복된 수를 고르지 않기 위한 코드만 수정해주면 되는 문제였습니다. 중복되는 수열을 또 출력하면 안되는 조건은 동일했습니다.

 

 

[소스코드]

 

#include<algorithm>
#include<iostream>
using namespace std;
 
int n, m, num[8], result[8];
 
// 입력받은 n개의 수들 중에서 골랐던 수열과 똑같은 수열을
// 고르지 않게 m개를 고르고 출력해주는 함수
void getResult(int count) {
 
    // m개를 골랐다면 출력합니다
    if (count == m) {
        for (int i = 0; i < m; i++)
            cout << result[i] << ' ';
        cout << '\n';
        return;
    }
    int prev_num = -1; // 바로 직전에 골랐던 수를 저장하는 변수
    
    for (int i = 0; i < n; i++) {
 
        // 중복되는 수열을 출력하지 않기위해 바로 직전에
        // 골랐던 수를 고르지 않게 합니다.
        if (prev_num != num[i]) {
            result[count] = num[i];
            prev_num = num[i];
            getResult(count + 1);
        }
    }
}
 
int main(void) {
 
    cin >> n >> m;
    for (int i = 0; i < n; i++)
        cin >> num[i];
 
    // 수열은 사전 순으로 출력해야 하므로
    // 입력을 받고 오름차순으로 정렬합니다
    sort(num, num + n);
 
    getResult(0);
    return 0;
}