DFS를 이용한, 순열(permutation) nPr 구하기 in C++

Purple·2021년 9월 13일
0

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

#include <bits/stdc++.h>

using namespace std;

int n, r;
int arr[101], ch[101], res[101];
void DFS(int Level) {
    if (Level == r + 1) {
        for (int i = 1; i <= r; i++) {
            cout << res[i] << ' ';
        }
        cout << '\n';
    }
    else {
        for (int i = 1; i <= n; i++) {
            if (ch[i] == 0) {
                ch[i] = 1;
                res[Level] = arr[i];
                DFS(Level + 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);
    return 0;
}

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

ex)
4 3
1 3 6 7

profile
안녕하세요.

0개의 댓글