백준 15655 N과 M (6)

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

15655번: N과 M (6)

1부터 n까지가 아닌, 주어진 숫자들을 이용하여 조합을 출력하는 문제이다.

#include <bits/stdc++.h>
using namespace std;
 
int n, m, a[10], ans[10];
 
void solve(int start, int idx) {
    if(idx == m) {
        for(int i = 0; i < m; i++)
            cout << ans[i] << ' ';
        cout << '\n';
        return;
    }
 
    for(int i = start; i < n; i++) {
        ans[idx] = a[i];
        solve(i + 1, idx + 1);
    }
}
 
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    cin >> n >> m;
    for(int i = 0; i < n; i++) cin >> a[i];
    sort(a, a + n);
    solve(0, 0);
    return 0;
}
 

아래코드는 시간복잡도를 O(2^n)으로 줄인 코드이다.

#include <bits/stdc++.h>
using namespace std;
 
int n, m, a[10], ans[10];
 
void solve(int start, int idx) {
    if(idx == m) {
        for(int i = 0; i < m; i++)
            cout << ans[i] << ' ';
        cout << '\n';
        return;
    }
    if(start >= n) return;
 
    ans[idx] = a[start];
    solve(start + 1, idx + 1);
    solve(start + 1, idx);
}
 
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    cin >> n >> m;
    for(int i = 0; i < n; i++) cin >> a[i];
    sort(a, a + n);
    solve(0, 0);
    return 0;
}