💻 C++ 기반

N과 M (9)
https://www.acmicpc.net/problem/15663

✔️ 주어지는 N개의 수 중에는 서로 같은 수가 존재할 수 있으므로 isUsed 배열에 저장할 때 값이 아닌, index로 저장해야 한다.

1. 백트래킹 코드

 #include <cstdio>
 #include <vector>
 #include <algorithm>

 #define MAX 9

 using namespace std;

 int N, M;
 vector<int> input;
 int output[MAX];
 bool isUsed[MAX];

 void func(int K)
 {
     if (K == M)
     {
         for (int i = 0; i < M; i++)
         {
             printf("%d ", output[i]);
         }
         printf("\n");
         return;
     }

     int prev = 0;
     for (int i = 0; i < N; i++)
     {
         if (!isUsed[i] && prev != input[i])
         {
             isUsed[i] = true;
             output[K] = input[i];
             prev = input[i];
             func(K + 1);
             isUsed[i] = false;
         }
     }
 }

 int main()
 {
     scanf("%d %d", &N, &M);
     for (int i = 0; i < N; i++)
     {
         int num;
         scanf("%d", &num);
         input.push_back(num);
     }
     sort(input.begin(), input.end());

     func(0);

     return 0;
 }

2. next_permutation 코드

#include <cstdio>
#include <vector>
#include <algorithm>

#define MAX 9

using namespace std;

int N, M;
vector<int> input;
vector<int> arr;
vector<vector<int> > output;

int main()
{
    scanf("%d %d", &N, &M);
    for (int i = 0; i < N; i++)
    {
        int num;
        scanf("%d", &num);
        input.push_back(num);
    }
    sort(input.begin(), input.end());

    for (int i = 0; i < M; i++)
    {
        arr.push_back(0);
    }
    for (int i = M; i < N; i++)
    {
        arr.push_back(1);
    }

    do
    {
        vector<int> temp;
        for (int i = 0; i < N; i++)
        {
            if (arr[i] == 0)
            {
                temp.push_back(input[i]);
            }
        }

        do
        {
            output.push_back(temp);
        } while (next_permutation(temp.begin(), temp.end()));
        
    } while (next_permutation(arr.begin(), arr.end()));

    sort(output.begin(), output.end());
    output.erase(unique(output.begin(), output.end()), output.end());

    for (int i = 0; i < output.size(); i++)
    {
        for (int j = 0; j < output[i].size(); j++)
        {
            printf("%d ", output[i][j]);
        }
        printf("\n");
    }

    return 0;
}