12101 : 1, 2, 3 더하기 2

풀이

지금 생각해보니 스택으로 짜도 될 것 같다.

코드

#include <stdio.h>
#include <vector>
using namespace std;

int N, K;
int res;
vector<int> ans;

bool go(int sum, int cnt) {
	if (sum == N) ++res;
	if (res == K) return true;
	for (int i = 1; i <= 3; ++i) {
		if (sum + i <= N) {
			if (go(sum + i, cnt + 1)) {
				ans.push_back(i);
				return true;
			}
		}
	}
	return false;
}

int main() {
	scanf("%d %d", &N, &K);

	go(0, 0);

	if (ans.empty()) printf("-1");
	else {
		printf("%d", ans[ans.size() - 1]);
		for (int i = ans.size() - 2; i >= 0; --i) printf("+%d", ans[i]);
	}

	return 0;
}

아무말

백준, 백준 온라인 저지, BOJ, Baekjoon Online Judge, C, C++, 씨, 씨쁠쁠, JAVA, algorithm, 자바, 알고리즘, 자료구조, 문제, 문제 풀이, 풀이

wookje.kwon's profile image

wookje.kwon

2017-02-23 12:41

Read more posts by this author