알고리즘

[C++] Boj 15655 N과 M (6)

(ꐦ •᷄ࡇ•᷅) 2025. 1. 22. 09:20

문제 링크

15655 N과 M (6)

 

접근

N개의 자연수 중, M개의 자연수를 뽑아 수열을 만든다. 이때 그 수열은 중복되는 수 조합이 아니어야 하고, 사전순으로 출력되어야 한다.

 

N과 M(5) 문제 풀이에서 조건 하나를 추가하여 해결하였다. (1, 7)이라는 수열이 있을 때 (7, 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;

        // 중복으로 뽑았는지 확인
        if (k > 0 && picked[k - 1] > numbers[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;
}