[백준 c++] 15654 N과 M

문제

https://www.acmicpc.net/problem/15654
N개의 자연수 중에서 M개를 고른 수열을 모두 출력하는 문제

---

풀이

조합을 재귀로 구현하는 문제다.

재귀함수로 조합을 구현하기위해 필요한 2가지 배열이 있다.

1. 방문 체크 배열 bool visited[]
2. 고른 조합을 담을 배열 arr

함수 작동 과정

1. 함수에서는 index 0~n까지 for문을 돌린다.
2. visited[i] = 1로 이미 방문처리된 원소라면 continue,
   visited[i] = 0이면 방문처리한 후 조합을 담을 배열 arr에 넣어준다.
3. 재귀 함수를 호출한다.
4. visited[i] = 0

---

코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int n, m, a, visited[9], arr[9];
vector<int> c;

void go(int idx)
{
    if (idx == m)
    {
        for (int i = 0; i < m; i++)
        {
            cout << arr[i] << " ";
        }
        cout << "\n";
        return;
    }

    for (int i = 0; i < n; i++)
    {
        if (visited[i])
            continue;
        visited[i] = 1;
        arr[idx] = c[i];
        go(idx + 1);
        visited[i] = 0;
    }
}

int main()
{
    cin >> n >> m;
    for (int i = 0; i < n; i++)
    {
        cin >> a;
        c.push_back(a);
    }

    sort(c.begin(), c.end());
    go(0);
}