[백준] 15566번: N과 M (12)

Solved.ac 실버2
https://www.acmicpc.net/problem/15666

설명

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

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

입력

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

출력

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

해결 Tip

알고리즘 분류

• 백트래킹

입력 받은 수들로 집합(Set)을 만들고 집합의 원소들로 M개의 수로 이루어진 조합(C)을 시도한다.

• Set : 원소들이 중복되지 않고, 정렬된 배열
• C : 확률과 통계에서 쓰는 조합 기호

코드

#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 outer[8];

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

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