#include <bits/stdc++.h>
using namespace std;
const int MAX_N = 1e6;
const long long INF = 1e18;
int a[MAX_N + 1], countLowerA[MAX_N + 1], minGroups[MAX_N + 1], maxGroups[MAX_N + 1];
long long sumA[MAX_N + 1], sumLowerA[MAX_N + 1], minSwaps[MAX_N + 1];
void computeDP( int n, long long cost ) {
for ( int i = 1; i <= n; i++ ) {
minSwaps[i] = INF;
minGroups[i] = n + 1;
maxGroups[i] = 0;
for ( int j = i - 1; j >= 0; j-- ) {
long long s = (minSwaps[j] + sumA[j]) + i * (-j);
if ( s < minSwaps[i] ) {
minSwaps[i] = s;
minGroups[i] = minGroups[j] + 1;
maxGroups[i] = maxGroups[j] + 1;
} else if ( s == minSwaps[i] ) {
minGroups[i] = min( minGroups[i], minGroups[j] + 1 );
maxGroups[i] = max( maxGroups[i], maxGroups[j] + 1 );
}
}
minSwaps[i] += i * countLowerA[i] - sumLowerA[i] + cost;
}
}
int main() {
int n, k, swaps;
cin >> n >> k;
int countB = 0, countA = 0;
swaps = 0;
for ( int i = 0; i < 2 * n; i++ ) {
char ch;
cin >> ch;
if ( ch == 'A' )
countA++;
else {
countB++;
swaps += max( 0, countB - countA );
a[countB] = max( countA, countB );
sumA[countB] = sumA[countB - 1] + a[countB];
countLowerA[a[countB]]++;
sumLowerA[a[countB]] += a[countB];
}
}
for ( int i = 1; i <= n; i++ ) {
countLowerA[i] += countLowerA[i - 1];
sumLowerA[i] += sumLowerA[i - 1];
}
long long leftCost = -1, rightCost = 1e12;
while ( rightCost - leftCost > 1 ) {
long long cost = (leftCost + rightCost) / 2;
computeDP( n, cost );
if ( minGroups[n] > k )
leftCost = cost;
else
rightCost = cost;
}
computeDP( n, rightCost );
cout << swaps + minSwaps[n] - min( k, maxGroups[n] ) * rightCost;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10588 KB |
Output is correct |
2 |
Correct |
2 ms |
10588 KB |
Output is correct |
3 |
Correct |
2 ms |
10584 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10704 KB |
Output is correct |
6 |
Correct |
2 ms |
10588 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
3 ms |
10588 KB |
Output is correct |
9 |
Correct |
2 ms |
10588 KB |
Output is correct |
10 |
Correct |
2 ms |
10588 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
2 ms |
10588 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
3 ms |
10588 KB |
Output is correct |
15 |
Correct |
2 ms |
10588 KB |
Output is correct |
16 |
Correct |
2 ms |
10588 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10588 KB |
Output is correct |
2 |
Correct |
2 ms |
10588 KB |
Output is correct |
3 |
Correct |
2 ms |
10584 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10704 KB |
Output is correct |
6 |
Correct |
2 ms |
10588 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
3 ms |
10588 KB |
Output is correct |
9 |
Correct |
2 ms |
10588 KB |
Output is correct |
10 |
Correct |
2 ms |
10588 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
2 ms |
10588 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
3 ms |
10588 KB |
Output is correct |
15 |
Correct |
2 ms |
10588 KB |
Output is correct |
16 |
Correct |
2 ms |
10588 KB |
Output is correct |
17 |
Correct |
3 ms |
10588 KB |
Output is correct |
18 |
Correct |
9 ms |
10720 KB |
Output is correct |
19 |
Correct |
9 ms |
10588 KB |
Output is correct |
20 |
Correct |
9 ms |
10588 KB |
Output is correct |
21 |
Correct |
9 ms |
10720 KB |
Output is correct |
22 |
Correct |
8 ms |
10840 KB |
Output is correct |
23 |
Correct |
9 ms |
10588 KB |
Output is correct |
24 |
Correct |
9 ms |
10588 KB |
Output is correct |
25 |
Correct |
9 ms |
10704 KB |
Output is correct |
26 |
Correct |
9 ms |
10584 KB |
Output is correct |
27 |
Correct |
9 ms |
10584 KB |
Output is correct |
28 |
Correct |
9 ms |
10708 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10588 KB |
Output is correct |
2 |
Correct |
2 ms |
10588 KB |
Output is correct |
3 |
Correct |
2 ms |
10584 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10704 KB |
Output is correct |
6 |
Correct |
2 ms |
10588 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
3 ms |
10588 KB |
Output is correct |
9 |
Correct |
2 ms |
10588 KB |
Output is correct |
10 |
Correct |
2 ms |
10588 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
2 ms |
10588 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
3 ms |
10588 KB |
Output is correct |
15 |
Correct |
2 ms |
10588 KB |
Output is correct |
16 |
Correct |
2 ms |
10588 KB |
Output is correct |
17 |
Correct |
3 ms |
10588 KB |
Output is correct |
18 |
Correct |
9 ms |
10720 KB |
Output is correct |
19 |
Correct |
9 ms |
10588 KB |
Output is correct |
20 |
Correct |
9 ms |
10588 KB |
Output is correct |
21 |
Correct |
9 ms |
10720 KB |
Output is correct |
22 |
Correct |
8 ms |
10840 KB |
Output is correct |
23 |
Correct |
9 ms |
10588 KB |
Output is correct |
24 |
Correct |
9 ms |
10588 KB |
Output is correct |
25 |
Correct |
9 ms |
10704 KB |
Output is correct |
26 |
Correct |
9 ms |
10584 KB |
Output is correct |
27 |
Correct |
9 ms |
10584 KB |
Output is correct |
28 |
Correct |
9 ms |
10708 KB |
Output is correct |
29 |
Correct |
631 ms |
10764 KB |
Output is correct |
30 |
Correct |
614 ms |
10768 KB |
Output is correct |
31 |
Correct |
617 ms |
10832 KB |
Output is correct |
32 |
Correct |
642 ms |
10588 KB |
Output is correct |
33 |
Correct |
674 ms |
10764 KB |
Output is correct |
34 |
Correct |
628 ms |
10832 KB |
Output is correct |
35 |
Correct |
611 ms |
10588 KB |
Output is correct |
36 |
Correct |
646 ms |
10764 KB |
Output is correct |
37 |
Correct |
623 ms |
10764 KB |
Output is correct |
38 |
Correct |
626 ms |
10588 KB |
Output is correct |
39 |
Correct |
627 ms |
10836 KB |
Output is correct |
40 |
Correct |
632 ms |
10764 KB |
Output is correct |
41 |
Correct |
646 ms |
10760 KB |
Output is correct |
42 |
Correct |
656 ms |
10588 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10588 KB |
Output is correct |
2 |
Correct |
2 ms |
10588 KB |
Output is correct |
3 |
Correct |
2 ms |
10584 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10704 KB |
Output is correct |
6 |
Correct |
2 ms |
10588 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
3 ms |
10588 KB |
Output is correct |
9 |
Correct |
2 ms |
10588 KB |
Output is correct |
10 |
Correct |
2 ms |
10588 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
2 ms |
10588 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
3 ms |
10588 KB |
Output is correct |
15 |
Correct |
2 ms |
10588 KB |
Output is correct |
16 |
Correct |
2 ms |
10588 KB |
Output is correct |
17 |
Correct |
3 ms |
10588 KB |
Output is correct |
18 |
Correct |
9 ms |
10720 KB |
Output is correct |
19 |
Correct |
9 ms |
10588 KB |
Output is correct |
20 |
Correct |
9 ms |
10588 KB |
Output is correct |
21 |
Correct |
9 ms |
10720 KB |
Output is correct |
22 |
Correct |
8 ms |
10840 KB |
Output is correct |
23 |
Correct |
9 ms |
10588 KB |
Output is correct |
24 |
Correct |
9 ms |
10588 KB |
Output is correct |
25 |
Correct |
9 ms |
10704 KB |
Output is correct |
26 |
Correct |
9 ms |
10584 KB |
Output is correct |
27 |
Correct |
9 ms |
10584 KB |
Output is correct |
28 |
Correct |
9 ms |
10708 KB |
Output is correct |
29 |
Correct |
631 ms |
10764 KB |
Output is correct |
30 |
Correct |
614 ms |
10768 KB |
Output is correct |
31 |
Correct |
617 ms |
10832 KB |
Output is correct |
32 |
Correct |
642 ms |
10588 KB |
Output is correct |
33 |
Correct |
674 ms |
10764 KB |
Output is correct |
34 |
Correct |
628 ms |
10832 KB |
Output is correct |
35 |
Correct |
611 ms |
10588 KB |
Output is correct |
36 |
Correct |
646 ms |
10764 KB |
Output is correct |
37 |
Correct |
623 ms |
10764 KB |
Output is correct |
38 |
Correct |
626 ms |
10588 KB |
Output is correct |
39 |
Correct |
627 ms |
10836 KB |
Output is correct |
40 |
Correct |
632 ms |
10764 KB |
Output is correct |
41 |
Correct |
646 ms |
10760 KB |
Output is correct |
42 |
Correct |
656 ms |
10588 KB |
Output is correct |
43 |
Execution timed out |
7103 ms |
13136 KB |
Time limit exceeded |
44 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
10588 KB |
Output is correct |
2 |
Correct |
2 ms |
10588 KB |
Output is correct |
3 |
Correct |
2 ms |
10584 KB |
Output is correct |
4 |
Correct |
2 ms |
10588 KB |
Output is correct |
5 |
Correct |
2 ms |
10704 KB |
Output is correct |
6 |
Correct |
2 ms |
10588 KB |
Output is correct |
7 |
Correct |
2 ms |
10588 KB |
Output is correct |
8 |
Correct |
3 ms |
10588 KB |
Output is correct |
9 |
Correct |
2 ms |
10588 KB |
Output is correct |
10 |
Correct |
2 ms |
10588 KB |
Output is correct |
11 |
Correct |
2 ms |
10588 KB |
Output is correct |
12 |
Correct |
2 ms |
10588 KB |
Output is correct |
13 |
Correct |
2 ms |
10588 KB |
Output is correct |
14 |
Correct |
3 ms |
10588 KB |
Output is correct |
15 |
Correct |
2 ms |
10588 KB |
Output is correct |
16 |
Correct |
2 ms |
10588 KB |
Output is correct |
17 |
Correct |
3 ms |
10588 KB |
Output is correct |
18 |
Correct |
9 ms |
10720 KB |
Output is correct |
19 |
Correct |
9 ms |
10588 KB |
Output is correct |
20 |
Correct |
9 ms |
10588 KB |
Output is correct |
21 |
Correct |
9 ms |
10720 KB |
Output is correct |
22 |
Correct |
8 ms |
10840 KB |
Output is correct |
23 |
Correct |
9 ms |
10588 KB |
Output is correct |
24 |
Correct |
9 ms |
10588 KB |
Output is correct |
25 |
Correct |
9 ms |
10704 KB |
Output is correct |
26 |
Correct |
9 ms |
10584 KB |
Output is correct |
27 |
Correct |
9 ms |
10584 KB |
Output is correct |
28 |
Correct |
9 ms |
10708 KB |
Output is correct |
29 |
Correct |
631 ms |
10764 KB |
Output is correct |
30 |
Correct |
614 ms |
10768 KB |
Output is correct |
31 |
Correct |
617 ms |
10832 KB |
Output is correct |
32 |
Correct |
642 ms |
10588 KB |
Output is correct |
33 |
Correct |
674 ms |
10764 KB |
Output is correct |
34 |
Correct |
628 ms |
10832 KB |
Output is correct |
35 |
Correct |
611 ms |
10588 KB |
Output is correct |
36 |
Correct |
646 ms |
10764 KB |
Output is correct |
37 |
Correct |
623 ms |
10764 KB |
Output is correct |
38 |
Correct |
626 ms |
10588 KB |
Output is correct |
39 |
Correct |
627 ms |
10836 KB |
Output is correct |
40 |
Correct |
632 ms |
10764 KB |
Output is correct |
41 |
Correct |
646 ms |
10760 KB |
Output is correct |
42 |
Correct |
656 ms |
10588 KB |
Output is correct |
43 |
Execution timed out |
7103 ms |
13136 KB |
Time limit exceeded |
44 |
Halted |
0 ms |
0 KB |
- |