15654-N과 M (5)

www.acmicpc.net/problem/15654

15654번: N과 M (5)

N개의 자연수와 자연수 M이 주어졌을 때, 길이가 M인 순열을 구하는 문제입니다. 여기서, N개의 자연수는 모두 다른 수입니다.

 

이전 N과 M (1, ..., 4) 문제는 N개의 자연수가 {1, ..., N}로 주어졌다면, 해당 문제는 임의의 자연수를 입력받고 그에 대한 순열을 구해야합니다. 이점을 유의하여 문제를 풀면 됩니다.

코드

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

int N;
int M;
int Ns[8];

int numbers[9] = {1,};
bool picked[9] = {false,};

void permutation(int len)
{
    if(len == M + 1)
    {
        for(int i = 1; i <= len - 1; i++)
        {
            cout << numbers[i] << " ";
        }
        cout << '\n';
        return;
    }

    for(int i = 0; i < N; i++)
    {
        if(picked[i])
            continue;
        
        picked[i] = true;
        numbers[len] = Ns[i];
        permutation(len + 1);
        picked[i] = false;
    }
}

int main()
{
    cin >> N >> M;
    for(int i = 0; i < N; i++)
        cin >> Ns[i];
    sort(Ns, Ns + N);
    permutation(1);
    return 0;
}