[백준] 15664번 : N과 M (10) (C++)

15664번 : N과 M (10)

---

문제)

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

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 비내림차순이어야 한다.
  • 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

 

 

 

입력 :

첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

 

 

 

출력 :

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

 

 

 

 

풀이)

 

// 15664. N과 N (10)
#include <iostream>
#include <vector>
#include <algorithm>
 
using namespace std;
 
int n, m;
vector<int> vec;
 
vector<int> arr;
int visited[100010];
 
void recur(int cur, int idx)
{
    // m자리 까지 모두 선택했다면 출력
    if (cur == m)
    {
        for (auto ele : arr)
        {
            cout << ele << " ";
        }
        cout << endl;
        return;
    }
 
    // 같은 수를 2번 이상 출력하지 못하게 하는 변수
    int temp = -1;
 
    for (int i = idx; i < n; i++)
    {
        // 이미 사용한 인자는 사용하지 못함
        if (visited[i]) continue;
        // 같은 수는 2번 이상 사용하지 못한다. (중복 방지)
        if (vec[i] == temp) continue;
 
        visited[i] = 1;
        temp = vec[i];
        arr[cur] = vec[i];
        recur(cur + 1, i);
        visited[i] = 0;
    }
}
 
int main()
{
    cin >> n >> m;
 
    int a;
    for (int i = 0; i < n; i++)
    {
        cin >> a;
        vec.push_back(a);
    }
 
    sort(vec.begin(), vec.end());
    arr.resize(m, 0);
 
    recur(0, 0);
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/15664 

15664번: N과 M (10)