Submission #906434

#	Time	Username	Problem	Language	Result	Execution time	Memory
906434		LucaIlie	Chorus (JOI23_chorus)	C++17	40 / 100	7031 ms	96960 KiB

chorus

#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 5e3;
int a[MAX_N + 1], sa[MAX_N + 1], minSwaps[MAX_N + 1][MAX_N + 1];

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 );
            sa[countB] = sa[countB - 1] + a[countB];
        }
    }

    for ( int i = 1; i <= n; i++ )
        minSwaps[i][0] = n * n;
    for ( int l = 1; l <= k; l++ ) {
        for ( int i = 1; i <= n; i++ ) {
            minSwaps[i][l] = n * n;
            int p = i;
            for ( int j = i - 1; j >= 0 && a[j + 1] >= i; j-- ) {
                minSwaps[i][l] = min( minSwaps[i][l], minSwaps[j][l - 1] );
                p = j;
            }
            for ( int j = p - 1; j >= 0; j-- )
                minSwaps[i][l] = min( minSwaps[i][l], minSwaps[j][l - 1] - i * j + sa[j] + i * p - sa[p] );
        }
    }

    cout << swaps + minSwaps[n][k];

    return 0;
}

• Subtask 1: 16 / 16
• Subtask 2: 24 / 24
• Subtask 3: 0 / 21
• Subtask 4: 0 / 26
• Subtask 5: 0 / 13