DFS를 활용한, 조합(combination) nCr 구하기 in C++

Purple·2021년 9월 13일
0

1. 특정 배열이 주어졌을 때, nCr 구하기

#include <bits/stdc++.h>

using namespace std;

int n, r;
int arr[101], ch[101], res[101];
void DFS(int Level, int start){
    if (Level == r + 1) {
        for (int i = 1; i <= r; i++) {
            cout << res[i] << ' ';
        }
        cout << '\n';
    }
    else {
        for (int i = start; i <= n; i++) {
            if (ch[i] == 0) {			// permutation일때 체크하는 것이지만, 그냥 통일감을 주기위해서 같은 흐름대로 짜봤다.
                ch[i] = 1;
                res[Level] = arr[i];
                DFS(Level + 1, i + 1);  // start+1하지 않도록 주의
                ch[i] = 0;
            }
        }
    }
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    freopen("input.txt", "rt", stdin);

    cin >> n >> r;
    for (int i = 1; i <= n; i++) cin >> arr[i];
    DFS(1, 1);
    return 0;
}

나같은 경우에는, permutation 이나 combination이나 트리형태로 그려서 표현하면 더 헷갈리지 않고 구현하기가 쉬웠다. 그래서 항상 간단하게나마 그림을 그리고 시작한다.

ex)
4 3
1 3 6 7

profile
안녕하세요.

0개의 댓글