Posts 12969 ABC
Post
Cancel

12969 ABC

12969 ABC

알고리즘(dp)

1
2
3
1. 현재 깊이 d, 이전에 나왔던 a의 개수, 이전에 나왔던 b의 개수, 이제까지의 k 수를 기준으로 dp를 만들어야함. k도 결과에 영향을 주기 때문.
2. 즉, dp[d][a][b][k] 를 선언하고, 이 상태가 검사되면 true로 지정하여 다시 검사하지않음.
3. 정답을 찾았을 때는 전역변수 got에 true값을 지정하여 바로 다른 모든걸 종료시키고 출력

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include <iostream>
#include <cstring>
using namespace std;

int N, K, num[30], ans[30];
bool save[31][31][31][450], got;

void DP(int d, int a, int b, int n);

int main() {
	cin >> N >> K;
	memset(num, -1, sizeof(num));
	memset(ans, -1, sizeof(ans));

	DP(0, 0, 0, 0);
	if (!got)
		cout << -1 << endl;
	else {
		for (int i = 0; i < N; i++) {
			if (ans[i] == 0)
				cout << 'A';
			else if (ans[i] == 1)
				cout << 'B';
			else if (ans[i] == 2)
				cout << 'C';
		}
	}
}

void DP(int d, int a, int b, int n) {
	if (d == N) {
		if (n == K) {
			got = true;
			memcpy(ans, num, sizeof(num));
		}
		return;
	}
	if (save[d][a][b][n])
		return;
	save[d][a][b][n] = true;

	if (got)
		return;

	num[d] = 0;
	DP(d + 1, a + 1, b, n);
	if (!got) {
		num[d] = 1;
		DP(d + 1, a, b + 1, n + a);
	}
	if (!got) {
		num[d] = 2;
		DP(d + 1, a, b, n + a + b);
	}
	num[d] = -1;
}
This post is licensed under CC BY 4.0 by the author.

12849 본대산책

1339 단어 수학

Comments powered by Disqus.