[백준] 15654번: N과 M (5)

Solved.ac 실버3
https://www.acmicpc.net/problem/15654

설명

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

• N개의 자연수 중에서 M개를 고른 수열

입력

첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

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

해결 Tip

알고리즘 분류

• 백트래킹

1. 재귀 함수를 이용해 수를 하나씩 선택해 배열이나 리스트에 저장하고, 그 개수가 M과 같아지면 출력한다.
2. 주의해야 할 점은 다음과 같다.
   • 중복된 수를 고르지 않을 것
   • 출력은 오름차순으로 할 것

코드

#include <bits/stdc++.h>
#define fastio ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
using namespace std;

int N, M;
int A[8], ans[8];
bool check[8];

void solve(int len) {
    if (len == M) {
        for (int i=0 ; i < len ; i++)
            cout << ans[i] << ' ';
        cout << '\n';
        return;
    }
    else {
        for (int i=0 ; i < N ; i++) {
            if (!check[i]) {
                check[i] = true;
                ans[len] = A[i];
                solve(len+1);
                check[i] = false;
            }
        }
    }
}

int main() {
    fastio;
    cin >> N >> M;
    for (int i=0 ; i < N ; i++)
        cin >> A[i];
    sort(A, A+N);
    solve(0);
	return 0;
}