(C++) 백준 15657번 - N과 M (8)

문제 링크 : https://www.acmicpc.net/problem/15657  

15657번: N과 M (8)

문제 풀이

백트래킹을 사용하여 풀이하였다.

입력받은 수들을 정렬한 후 배열에 들어올 수가 이전에 배열에 삽입된 수보다 작지 않도록 조건만 달아주면 쉽게 풀 수 있다.

#include <iostream>
#include <algorithm>

using namespace std;

int n, m;
int arr[10]; // 입력값들을 저장하는 배열
int result[10]; // 출력 순서를 저장할 배열
int isUsed[10];

void nNm(int idx){
    if(idx == m){ // base condition
            for(int i = 0; i < m; i++){
                cout << result[i] << " ";
            }
            cout << "\n";
            return;
        }
    for(int i = 0; i < n; i++){
        if(arr[i] < result[idx-1]) continue; // 들어올 수가 이전에 배열에 삽입된 수보다 작으면 그 수는 스킵
        result[idx] = arr[i]; // 배열에 삽입하고
        nNm(idx+1); // 백트래킹
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> m;

    for(int i = 0; i < n; ++i){
        int tmp;
        cin >> tmp;
        arr[i] = tmp;
    }
    sort(arr, arr+n);
    nNm(0);    
}