[백준] 15654번

💻 C++ 코드

N과 M (5)
https://www.acmicpc.net/problem/15654

1. 백트래킹 코드

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

#define MAX 9
#define MAX_NUM 10001

using namespace std;

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

void func(int K)
{
    if (K == M)
    {
        for (int i = 0; i < M; i++)
        {
            printf("%d ", output[i]);
        }
        printf("\n");
        return;
    }
    for (int i = 0; i < N; i++)
    {
        if (!isUsed[input[i]])
        {
            isUsed[input[i]] = true;
            output[K] = input[i];
            func(K + 1);
            isUsed[input[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 코드

✔️ next_permutation의 경우 중복된 수열을 계산해주지 않으므로 우선 M개만큼 뽑은 다음, 새롭게 next_permutation을 돌려 순열을 구해줘야 함

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

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());

    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;
}

좋은 웹페이지 즐겨찾기