[C++] Boj 15654 N과 M (5)

문제 링크

15654 N과 M (5)

 

접근

기존 N과 M (1) 문제에서 입력을 받는 케이스만 추가된 문제이다. 이때 중요한 점은 사전순으로 수열을 출력해야 하므로, 배열에 수들을 입력받고 정렬해 주었다.

 

코드

#include <iostream>
#include <algorithm>
using namespace std;

// N개 중에서, M개 뽑음
int N, M;

// n번째로 뽑은 수를 저장
int picked[10];

// 숫자 모음
int numbers[10];

// 방문했는지
bool isUsed[10];

// k 번째로 저장할 수를 탐색 
void Solve(int k)
{
    if (k == M)
    {
        for (int i = 0; i < M; i++)
            cout << picked[i] << ' ';
        cout << '\n';
        return;
    }

    for (int i = 0; i < N; i++)
    {
        if (isUsed[i]) continue;

        picked[k] = numbers[i];
        isUsed[i] = true;
        Solve(k + 1);
        isUsed[i] = false;
    }
}

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

    cin >> N >> M;
    for (int i = 0; i < N; i++)
        cin >> numbers[i];
   
    // 사전순으로 찾기 위해 배열을 정렬
    sort(numbers, numbers + N);

    Solve(0);

    return 0;
}