[백준] 15657번: N과 M (8)

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

설명

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

• N개의 자연수는 모두 다른 수이다.
• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 고른 수열은 비내림차순이어야 한다.
• 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

입력

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

출력

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

해결 Tip

알고리즘 분류

• 백트래킹

1. 먼저 주어진 수를 정렬하고, 재귀함수를 이용해 문제를 푼다
2. 이때, 재귀함수에서 다음에 선택할 수의 범위를 [현재 고른 수의 인덱스, 인덱스 N-1]로 지정한다.

코드

#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];

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

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