알고리즘/알고리즘 문제 풀이

(C++) 백준 15650번 - N과 M (2)

용꿀 2023. 4. 9. 19:56

문제 링크 : https://www.acmicpc.net/problem/15650

 

15650번: N과 M (2)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

문제 풀이

백트래킹을 사용한 풀이이다. 백트래킹을 하면서 이전에 들어간 수보다 들어갈 수가 더 적다면 그 경우는 건너뛰는 식으로 구현한다.

#include <iostream>

using namespace std;

int n, m;
int arr[10];
int isUsed[10];

void nNm(int idx){
    if(idx == m){ // base condition
            for(int i = 0; i < m; i++){
                cout << arr[i] << " ";
            }
            cout << "\n";
            return;
        }
    for(int i = 1; i <= n; i++){
        if(i < arr[idx-1]) continue; // 오름차순이 아니면 다음 수로 넘기기
        if(!isUsed[i]){ // 사용하지 않은 수면
            arr[idx] = i; // 배열에 삽입하고
            isUsed[i] = 1; // 사용되었다고 표시
            nNm(idx+1); // 백트래킹
            isUsed[i] = 0; // 사용 완료
        }
    }
}

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

    cin >> n >> m;

    nNm(0);
}