제출 #1346935

#제출 시각아이디문제언어결과실행 시간메모리
1346935MMihalevChorus (JOI23_chorus)C++20
0 / 100
1 ms356 KiB
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int MAX_N=5e3+3;
int dp[MAX_N][MAX_N];
int cost[MAX_N][MAX_N];
int p[MAX_N];
int n,k;
string s;
int main ()
{
    ios_base::sync_with_stdio(0);
    cin.tie(NULL);
    cout.tie(NULL);
    
    cin>>n>>k;
    cin>>s;

    int pos=0;
    int cnt=0;
    for(int i=0;i<2*n;i++)
    {
        if(s[i]=='A')
        {
            pos++;
            p[pos]=cnt;
        }
        else cnt++;
    }

    for(int pr=0;pr<=n;pr++)
    {
        for(int i=1;i<=n;i++)
        {
            cost[i][pr]=max(0,p[i]-pr)+cost[i-1][pr];
        }
    }

    for(int i=0;i<=n;i++)
    {
        for(int gr=0;gr<=n;gr++)
        {
            dp[i][gr]=1e9;
        }
    }

    dp[0][0]=0;
    for(int i=1;i<=n;i++)
    {
        for(int gr=1;gr<=i;gr++)
        {
            for(int j=0;j<i;j++)
            {
                if(dp[j][gr-1]==1e9)continue;

                if(p[j]>j)continue;

                dp[i][gr]=min(dp[i][gr],
                              dp[j][gr-1]+cost[i][j]-cost[j][j]);
            }
        }
    }

    int ans=1e9;
    for(int gr=1;gr<=k;gr++)
    {
        ans=min(ans,dp[n][gr]);
    }

    cout<<ans<<"\n";

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...