Wannafly挑战赛12 C.删除子串（前缀和优化dp）

题意：

解法：

因为只有当a[i]!=a[i+1]的时候变化,
所以容易发现删除一定是删除abbbba中的所有b,
或者是删除baaaaab中的所有a,即删除操作是一段一段操作的,
因此可以将串先分为若干段,相邻段的字符种类不同.

令d[i][j]表示前i段中,保留第i段,且变化次数为j能留下的最大长度,
转移方程:
1.d[i][j]=max(d[k][j-1])+a[i],
其中a[i]表示第i段的长度,k是[1,i-1]中与i颜色不同的段的位置.
max(d[k][j-1])可以用前缀和优化.

2.d[i][j]=max(d[k][j])+a[i],
其中a[i]表示第i段的长度,k是[1,i-1]中与i颜色相同的段的位置.
max(d[k][j])可以用前缀和优化.

算法复杂度O(n*m)

code：

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=2e6+5;
int d[maxm][15];
int ma[2][15];
char s[maxm];
int a[maxm];
int n,m;
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    cin>>n>>m;
    cin>>(s+1);
    int tot=0;
    for(int i=1;i<=n;i++){
        int j=i;
        while(j<=n&&s[j]==s[i])j++;
        j--;
        if(tot==0&&s[i]=='b'){//第一段必须为a,所以这一段不能保留.
            i=j;
            continue;
        }else{
            tot++;
            a[tot]=j-i+1;
            i=j;
        }
    }
    if(!tot)return cout<<0<<endl,0;//没有a
    for(int i=0;i<=tot;i++){
        for(int j=0;j<=m;j++){
            d[i][j]=-1e9;
        }
    }
    for(int j=0;j<=m;j++){
        ma[0][j]=ma[1][j]=-1e9;
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        if(i%2==1){//作为开头第一段(i为奇数表示是以字符'a'填充的段).
            d[i][0]=a[i];
            if(ma[1][0]>=0){
                d[i][0]=max(d[i][0],ma[1][0]+a[i]);
            }
        }
        for(int j=1;j<=m&&j<i;j++){
            if(ma[!(i%2)][j-1]>=0){//接在不同颜色的段后面
                d[i][j]=max(d[i][j],ma[!(i%2)][j-1]+a[i]);
            }
            if(ma[i%2][j]>=0){//接在相同颜色的段后面
                d[i][j]=max(d[i][j],ma[i%2][j]+a[i]);
            }
            ans=max(ans,d[i][j]);
        }
        for(int j=0;j<=m&&j<i;j++){
            ma[i%2][j]=max(ma[i%2][j],d[i][j]);
        }
    }
    cout<<ans<<endl;
    return 0;
}

---