제출 #1346941

#제출 시각아이디문제언어결과실행 시간메모리
1346941MMihalevChorus (JOI23_chorus)C++20
61 / 100
943 ms196316 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 gr;
int solveid(int l,int r,int posl,int posr)
{
    int mid=(l+r)/2;
    int i=mid;
    int opt;
    for(int j=posl;j<min(i,posr+1);j++)
    {
        if(dp[j][gr-1]==1e9)continue;
        int cur=dp[j][gr-1]+cost[i][j]-cost[j][j];

        if(cur<=dp[i][gr])
        {
            dp[i][gr]=cur;
            opt=j;
        }
    }
    return opt;
}
void solve(int l,int r,int posl,int posr)
{
    if(l>r)return;
    int mid=(l+r)/2;
    int opt=solveid(l,r,posl,posr);
    solve(l,mid-1,posl,opt);
    solve(mid+1,r,opt,posr);
}
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 j=0;j<=n;j++)dp[i][j]=1e9;
    }

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

    int ans=dp[n][1];
    for(gr=2;gr<=k;gr++)
    {
        solve(gr,n,gr-1,n);
        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...