본문으로 바로가기

[BOJ 15654] N과 M (5) (C++)

category Algorithms/Backtracking 2021. 9. 18. 13:04

N과 M(5) Silver 3

문제

전체 문제 보기

 

15654번: N과 M (5)

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

www.acmicpc.net

접근법

이 문제는 BOJ 15649 번 문제와 유사합니다. 그래서 접근법은 BOJ 15649 문제 풀이를 참고해 주세요.

전체 구현

#include <iostream>
#include <vector>
#include <algorithm>

#define endl '\n'
using namespace std;
void Backtracking(const vector<int>& arr, vector<int>& temp, vector<bool>& visited, const int M)
{
    if (temp.size() == M)
    {
        for (int n : temp)
            cout << n << " ";
        cout << endl;

        return;
    }

    for (int i = 0; i < arr.size(); ++i)
    {
        if (visited[i]) continue;

        temp.push_back(arr[i]);
        visited[i] = true;
        Backtracking(arr, temp, visited, M);
        temp.pop_back();
        visited[i] = false;
    }
}
int main() {
    // 입출력 성능 향상을 위한 설정
    ios_base::sync_with_stdio(false);
    cout.tie(NULL);
    cin.tie(NULL);

    int N, M; // (1 ≤ M ≤ N ≤ 8)
    cin >> N >> M;

    vector<int> arr(N);
    vector<int> temp;
    vector<bool> visited(N);

    for (int i = 0; i < arr.size(); ++i)
    {
        cin >> arr[i];
    }

    sort(arr.begin(), arr.end());

    Backtracking(arr, temp, visited, M);


    return 0;
}