#include <bits/stdc++.h>
//#define int long long
//#pragma GCC optimize("Ofast")
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4")
#define file(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);
#define forev(i, b, a) for(int i = (b); i >= (a); --i)
#define forn(i, a, b) for(int i = (a); i <= (b); ++i)
#define all(x) x.begin(), x.end()
#define sz(s) (int)s.size()
#define pb push_back
#define ppb pop_back
#define mp make_pair
#define s second
#define f first
using namespace std;
typedef pair < long long, long long > pll;
typedef pair < int, int > pii;
typedef unsigned long long ull;
typedef vector < pii > vpii;
typedef vector < int > vi;
typedef long double ldb;
typedef long long ll;
typedef double db;
const int dx[] = {1, -1, 0, 0}, dy[] = {0, 0, 1, -1}, block = 555;
const int inf = 1e9, maxn = 4e5 + 148, mod = 1e9 + 7, N = 411;
const pii base = mp(1171, 3307), Mod = mp(1e9 + 7, 1e9 + 9);
const db eps = 1e-12, pi = 3.14159265359;
const ll INF = 1e18;
int n, cnt[N][3], dp[N][N][N][3], c[3];
vi pos[3];
string s;
inline int get (int l, int r, int i) {
if (l > r)
return 0;
return cnt[r][i] - cnt[l - 1][i];
}
main () {
cin >> n >> s;
s = "#" + s;
memset(dp, -1, sizeof(dp));
for (int i = 1; i <= n; ++i) {
for (int j = 0; j < 3; ++j)
cnt[i][j] = cnt[i - 1][j];
if (s[i] == 'R') {
++cnt[i][0];
pos[0].pb(i);
} else if (s[i] == 'Y') {
++cnt[i][1];
pos[1].pb(i);
} else {
++cnt[i][2];
pos[2].pb(i);
}
}
int cnt0 = cnt[n][0], cnt1 = cnt[n][1], cnt2 = cnt[n][2];
for (int i = 0; i <= cnt0; ++i)
for (int j = 0; j <= cnt1; ++j)
for (int x = 0; x <= cnt2; ++x)
for (int lst = 0; lst < 3; ++lst)
dp[i][j][x][lst] = inf;
for (int lst = 0; lst < 3; ++lst)
dp[0][0][0][lst] = 0;
for (c[0] = 0; c[0] <= cnt0; ++c[0])
for (c[1] = 0; c[1] <= cnt1; ++c[1])
for (c[2] = 0; c[2] <= cnt2; ++c[2])
for (int lst = 0; lst < 3; ++lst) {
int cost = 0, c0 = c[0] - !lst, c1 = c[1] - (lst == 1), c2 = c[2] - (lst == 2);
if (c1 < 0 || c0 < 0 || c2 < 0)
continue;
if (lst && c[0])
cost += get(pos[lst][c[lst] - 1], pos[0][c[0] - 1], 0);
if (lst != 1 && c[1])
cost += get(pos[lst][c[lst] - 1], pos[1][c[1] - 1], 1);
if (lst != 2 && c[2])
cost += get(pos[lst][c[lst] - 1], pos[2][c[2] - 1], 2);
if (lst)
dp[c[0]][c[1]][c[2]][lst] = min(dp[c[0]][c[1]][c[2]][lst], dp[c0][c1][c2][0] + cost);
if (lst != 1)
dp[c[0]][c[1]][c[2]][lst] = min(dp[c[0]][c[1]][c[2]][lst], dp[c0][c1][c2][1] + cost);
if (lst != 2)
dp[c[0]][c[1]][c[2]][lst] = min(dp[c[0]][c[1]][c[2]][lst], dp[c0][c1][c2][2] + cost);
}
int ans = min({dp[cnt0][cnt1][cnt2][0], dp[cnt0][cnt1][cnt2][1], dp[cnt0][cnt1][cnt2][2]});
if (ans == inf)
ans = -1;
cout << ans << endl;
}
Compilation message
joi2019_ho_t3.cpp:52:7: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
main () {
^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
405 ms |
815480 KB |
Output is correct |
2 |
Correct |
408 ms |
815608 KB |
Output is correct |
3 |
Correct |
400 ms |
815484 KB |
Output is correct |
4 |
Correct |
410 ms |
815608 KB |
Output is correct |
5 |
Correct |
435 ms |
815480 KB |
Output is correct |
6 |
Correct |
405 ms |
815584 KB |
Output is correct |
7 |
Correct |
436 ms |
815608 KB |
Output is correct |
8 |
Correct |
418 ms |
815608 KB |
Output is correct |
9 |
Correct |
442 ms |
815608 KB |
Output is correct |
10 |
Correct |
411 ms |
815736 KB |
Output is correct |
11 |
Correct |
429 ms |
815608 KB |
Output is correct |
12 |
Correct |
412 ms |
815480 KB |
Output is correct |
13 |
Correct |
455 ms |
815608 KB |
Output is correct |
14 |
Correct |
432 ms |
815608 KB |
Output is correct |
15 |
Correct |
417 ms |
815480 KB |
Output is correct |
16 |
Correct |
437 ms |
815456 KB |
Output is correct |
17 |
Correct |
414 ms |
815568 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
405 ms |
815480 KB |
Output is correct |
2 |
Correct |
408 ms |
815608 KB |
Output is correct |
3 |
Correct |
400 ms |
815484 KB |
Output is correct |
4 |
Correct |
410 ms |
815608 KB |
Output is correct |
5 |
Correct |
435 ms |
815480 KB |
Output is correct |
6 |
Correct |
405 ms |
815584 KB |
Output is correct |
7 |
Correct |
436 ms |
815608 KB |
Output is correct |
8 |
Correct |
418 ms |
815608 KB |
Output is correct |
9 |
Correct |
442 ms |
815608 KB |
Output is correct |
10 |
Correct |
411 ms |
815736 KB |
Output is correct |
11 |
Correct |
429 ms |
815608 KB |
Output is correct |
12 |
Correct |
412 ms |
815480 KB |
Output is correct |
13 |
Correct |
455 ms |
815608 KB |
Output is correct |
14 |
Correct |
432 ms |
815608 KB |
Output is correct |
15 |
Correct |
417 ms |
815480 KB |
Output is correct |
16 |
Correct |
437 ms |
815456 KB |
Output is correct |
17 |
Correct |
414 ms |
815568 KB |
Output is correct |
18 |
Correct |
410 ms |
815476 KB |
Output is correct |
19 |
Correct |
403 ms |
815716 KB |
Output is correct |
20 |
Correct |
405 ms |
815484 KB |
Output is correct |
21 |
Correct |
405 ms |
815448 KB |
Output is correct |
22 |
Correct |
402 ms |
815480 KB |
Output is correct |
23 |
Correct |
400 ms |
815608 KB |
Output is correct |
24 |
Correct |
402 ms |
815608 KB |
Output is correct |
25 |
Correct |
408 ms |
815480 KB |
Output is correct |
26 |
Correct |
418 ms |
815460 KB |
Output is correct |
27 |
Correct |
405 ms |
815608 KB |
Output is correct |
28 |
Correct |
403 ms |
815608 KB |
Output is correct |
29 |
Correct |
445 ms |
815480 KB |
Output is correct |
30 |
Correct |
415 ms |
815608 KB |
Output is correct |
31 |
Correct |
409 ms |
815480 KB |
Output is correct |
32 |
Correct |
413 ms |
815736 KB |
Output is correct |
33 |
Correct |
407 ms |
815484 KB |
Output is correct |
34 |
Correct |
398 ms |
815480 KB |
Output is correct |
35 |
Correct |
403 ms |
815480 KB |
Output is correct |
36 |
Correct |
412 ms |
815548 KB |
Output is correct |
37 |
Correct |
403 ms |
815444 KB |
Output is correct |
38 |
Correct |
407 ms |
815464 KB |
Output is correct |
39 |
Correct |
407 ms |
815484 KB |
Output is correct |
40 |
Correct |
419 ms |
815608 KB |
Output is correct |
41 |
Correct |
419 ms |
815608 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
398 ms |
815480 KB |
Output is correct |
2 |
Correct |
407 ms |
815612 KB |
Output is correct |
3 |
Correct |
401 ms |
815448 KB |
Output is correct |
4 |
Correct |
409 ms |
815504 KB |
Output is correct |
5 |
Correct |
436 ms |
815480 KB |
Output is correct |
6 |
Correct |
403 ms |
815608 KB |
Output is correct |
7 |
Correct |
442 ms |
815608 KB |
Output is correct |
8 |
Correct |
409 ms |
815460 KB |
Output is correct |
9 |
Correct |
403 ms |
815512 KB |
Output is correct |
10 |
Correct |
414 ms |
815480 KB |
Output is correct |
11 |
Correct |
414 ms |
815480 KB |
Output is correct |
12 |
Correct |
433 ms |
815500 KB |
Output is correct |
13 |
Correct |
408 ms |
815456 KB |
Output is correct |
14 |
Correct |
413 ms |
815484 KB |
Output is correct |
15 |
Correct |
435 ms |
815460 KB |
Output is correct |
16 |
Correct |
417 ms |
815480 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
405 ms |
815480 KB |
Output is correct |
2 |
Correct |
408 ms |
815608 KB |
Output is correct |
3 |
Correct |
400 ms |
815484 KB |
Output is correct |
4 |
Correct |
410 ms |
815608 KB |
Output is correct |
5 |
Correct |
435 ms |
815480 KB |
Output is correct |
6 |
Correct |
405 ms |
815584 KB |
Output is correct |
7 |
Correct |
436 ms |
815608 KB |
Output is correct |
8 |
Correct |
418 ms |
815608 KB |
Output is correct |
9 |
Correct |
442 ms |
815608 KB |
Output is correct |
10 |
Correct |
411 ms |
815736 KB |
Output is correct |
11 |
Correct |
429 ms |
815608 KB |
Output is correct |
12 |
Correct |
412 ms |
815480 KB |
Output is correct |
13 |
Correct |
455 ms |
815608 KB |
Output is correct |
14 |
Correct |
432 ms |
815608 KB |
Output is correct |
15 |
Correct |
417 ms |
815480 KB |
Output is correct |
16 |
Correct |
437 ms |
815456 KB |
Output is correct |
17 |
Correct |
414 ms |
815568 KB |
Output is correct |
18 |
Correct |
410 ms |
815476 KB |
Output is correct |
19 |
Correct |
403 ms |
815716 KB |
Output is correct |
20 |
Correct |
405 ms |
815484 KB |
Output is correct |
21 |
Correct |
405 ms |
815448 KB |
Output is correct |
22 |
Correct |
402 ms |
815480 KB |
Output is correct |
23 |
Correct |
400 ms |
815608 KB |
Output is correct |
24 |
Correct |
402 ms |
815608 KB |
Output is correct |
25 |
Correct |
408 ms |
815480 KB |
Output is correct |
26 |
Correct |
418 ms |
815460 KB |
Output is correct |
27 |
Correct |
405 ms |
815608 KB |
Output is correct |
28 |
Correct |
403 ms |
815608 KB |
Output is correct |
29 |
Correct |
445 ms |
815480 KB |
Output is correct |
30 |
Correct |
415 ms |
815608 KB |
Output is correct |
31 |
Correct |
409 ms |
815480 KB |
Output is correct |
32 |
Correct |
413 ms |
815736 KB |
Output is correct |
33 |
Correct |
407 ms |
815484 KB |
Output is correct |
34 |
Correct |
398 ms |
815480 KB |
Output is correct |
35 |
Correct |
403 ms |
815480 KB |
Output is correct |
36 |
Correct |
412 ms |
815548 KB |
Output is correct |
37 |
Correct |
403 ms |
815444 KB |
Output is correct |
38 |
Correct |
407 ms |
815464 KB |
Output is correct |
39 |
Correct |
407 ms |
815484 KB |
Output is correct |
40 |
Correct |
419 ms |
815608 KB |
Output is correct |
41 |
Correct |
419 ms |
815608 KB |
Output is correct |
42 |
Correct |
398 ms |
815480 KB |
Output is correct |
43 |
Correct |
407 ms |
815612 KB |
Output is correct |
44 |
Correct |
401 ms |
815448 KB |
Output is correct |
45 |
Correct |
409 ms |
815504 KB |
Output is correct |
46 |
Correct |
436 ms |
815480 KB |
Output is correct |
47 |
Correct |
403 ms |
815608 KB |
Output is correct |
48 |
Correct |
442 ms |
815608 KB |
Output is correct |
49 |
Correct |
409 ms |
815460 KB |
Output is correct |
50 |
Correct |
403 ms |
815512 KB |
Output is correct |
51 |
Correct |
414 ms |
815480 KB |
Output is correct |
52 |
Correct |
414 ms |
815480 KB |
Output is correct |
53 |
Correct |
433 ms |
815500 KB |
Output is correct |
54 |
Correct |
408 ms |
815456 KB |
Output is correct |
55 |
Correct |
413 ms |
815484 KB |
Output is correct |
56 |
Correct |
435 ms |
815460 KB |
Output is correct |
57 |
Correct |
417 ms |
815480 KB |
Output is correct |
58 |
Correct |
482 ms |
815608 KB |
Output is correct |
59 |
Correct |
491 ms |
815624 KB |
Output is correct |
60 |
Correct |
486 ms |
815484 KB |
Output is correct |
61 |
Correct |
472 ms |
815736 KB |
Output is correct |
62 |
Correct |
411 ms |
815480 KB |
Output is correct |
63 |
Correct |
414 ms |
815468 KB |
Output is correct |
64 |
Correct |
447 ms |
815608 KB |
Output is correct |
65 |
Correct |
454 ms |
815632 KB |
Output is correct |
66 |
Correct |
474 ms |
815604 KB |
Output is correct |
67 |
Correct |
487 ms |
815684 KB |
Output is correct |
68 |
Correct |
473 ms |
815484 KB |
Output is correct |
69 |
Correct |
478 ms |
815608 KB |
Output is correct |
70 |
Correct |
481 ms |
815684 KB |
Output is correct |
71 |
Correct |
479 ms |
815456 KB |
Output is correct |
72 |
Correct |
435 ms |
815528 KB |
Output is correct |
73 |
Correct |
418 ms |
815608 KB |
Output is correct |