백준 12101번 1, 2, 3 더하기 2 C++

https://www.acmicpc.net/problem/12101

12101번: 1, 2, 3 더하기 2

 

dfs

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, k;
int cnt = 1;
bool ans = false;

vector<int> nums;

void Dfs(int sum) {
    if (sum > n) return;
    if (sum == n) {
        if (cnt++ == k) {
            ans = true;
            for (int i = 0; i < nums.size(); i++) {
                if (i != 0) cout << '+';
                cout << nums[i];
            }
        }
        return;
    }
    nums.push_back(1);
    Dfs(sum + 1);
    nums.pop_back();

    nums.push_back(2);
    Dfs(sum + 2);
    nums.pop_back();

    nums.push_back(3);
    Dfs(sum + 3);
    nums.pop_back();
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> k;
    Dfs(0);
    if (ans == false) cout << -1;
}