[백준/백트래킹/C++] 15650번 N과 M (2) (조합)

https://www.acmicpc.net/problem/15650

15650번: N과 M (2)

#include <bits/stdc++.h>

using namespace std;

int n, m;
int arr[9];
bool isused[9];

void backtrack(int k){ // 현재 k개 까지 수를 택했음
    if(k==m){
        for(int i=0; i<m; i++)
            cout << arr[i] << ' ';
        cout << '\n';
        return;
    }
    int st = 1; // 시작지점, k=0일 때에는 st = 1
    if(k!=0) st = arr[k-1] + 1; // k!=0 일 경우 st = arr[k-1]+1
    for(int i=st; i<=n; i++){
        if(!isused[i]){
            arr[k] = i;
            isused[i] = 1;
            backtrack(k+1);
            isused[i] = 0;
        }
    }
}

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> m;

    backtrack(0);

    return 0;
}

 

next_permutation() 사용

#include <bits/stdc++.h>

using namespace std;

int n, m;
vector<int> a;


int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> m;
    // m의 개수만큼 0을 넣어줌
    for(int i=0; i<n; i++) a.push_back(i<m? 0: 1);
    do{
        for(int i=0; i<n; i++)
            if(a[i] == 0) cout << i+1 << ' ';
        cout << '\n';
    }while(next_permutation(a.begin(), a.end()));

    return 0;
}