본문 바로가기
etc

[코드트리 조별과제] 3주차

by fafaya 2024. 8. 2.

[백트래킹] 특정 조건에 맞게 k개 중에 1개를 n번 뽑기

https://www.codetree.ai/missions/2/problems/n-permutations-of-k-with-repetition-under-constraint?&utm_source=clipboard&utm_medium=text

#include <iostream>
#include <vector>
using namespace std;

int k, n, cnt;
vector<int> v;

void printA(){
    for(int i: v) cout << i << " ";
    cout << "\n";
}

void choose(int num){
    if(num == n){
        printA();
        return;
    }
    int len = v.size();
    for(int i=1; i<=k; i++){
        if(num <=1 || v.back()!=i || v[len-2]!=i ){
            v.push_back(i);
            choose(num+1);
            v.pop_back();
        }
    }
}

int main() {
    cin >> k >> n;
    choose(0);


    return 0;
}

'etc' 카테고리의 다른 글

[코드트리 조별과제] 2주차  (1) 2024.07.26
[코드트리 조별과제] 1주차  (0) 2024.07.20

댓글