알고리즘/알고리즘 문제 풀이

(C++) 백준 15656번 - N과 M (7)

용꿀 2023. 4. 9. 23:14

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

 

15656번: N과 M (7)

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

www.acmicpc.net

문제 풀이

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

아주 간단한 백트래킹 문제로 N과 M (3) 문제처럼 특별한 조건이 없기에 다른 백트래킹 문제에 익숙하다면 어렵지 않게 풀 수 있는 문제이다.

입력받은 수들을 정렬한 후에 N과 M (3) 문제와 같이 풀이하면 된다.

#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++){
        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);    
}