wannafly 12 删除子串 dp

原文：https://www.cnblogs.com/genius777/p/8636661.html

删除子串


题目描述 
给你一个长度为n且由a和b组成的字符串，你可以删除其中任意的部分（可以不删），使得删除后的子串“变化”次数小于等于m次且最长。
变化：如果a[i]!=a[i+1]则为一次变化。（且新的字符串的首字母必须是'a'）
如果初始串全为b，则输出0。
输入描述:
第一行输入两个数n,m。（1 <= n <= 105,0 <= m <= 10）
第二行输入一行长度为n且由a和b组成的字符串
输出描述:
输出一个数字表示最长长度
示例1
输入
8 2
aabbabab
输出
6
说明
原串可以变成aabbbb，只改变了一次，且长度最长。


 


【题目分析】

可以参考0-1背包

这道题来自于牛客网的一场比赛，昨天第一次注册牛客网，打了一场比赛，然后就GG了，下来想了一下这道题，发现自己写得已经离AC很接近了，然后就把代码补上吧。


首先可以发现这道题其实算一道动规题，当然同机房的几位大佬又讲一些好似很厉害，很优秀的算法，不过这里就先写动规吧。。。


找状态转移方程是动规题的关键，由于这道题有“只由a、b组成的字符串”这条妙妙的性质，我们就可以利用这个妙妙的性质来写转移方程。


维护一个dp数组——>dp[i][j][k]。


[i]——>维护到了第几个字符


[j]——>表示我们使用了几次变化


[k]——>这一项只有两个，表示这次提取后的子串的末尾是a还是b。（这样就便于状态转移）

然后就可以写出状态转移方程

            if(j==0)
            {
                if(st[i]=='b')
                dp[i][j][1]=dp[i-1][j][1];//因为不用变化且字符串首项一定是a所以b不能被提取 
                else
                dp[i][j][1]=dp[i-1][j][1]+1;//是a就可以提取了 
                ans=ans>dp[i][j][1]?ans:dp[i][j][1];
            }
            else
            {
                if(st[i]=='a')//当前字符是a 
                {
                    dp[i][j][1]=max(dp[i-1][j-1][2]+1,dp[i-1][j][1]+1);//表示要选这个a
                    //选择的方式可以有两个，一个是用一次变化从b到a，或者不用变化从a到a 
                    dp[i][j][2]=max(dp[i][j-1][1],dp[i-1][j][2]);//表示不选这个a 
                }
                else
                {
                    dp[i][j][1]=max(dp[i-1][j-1][2],dp[i-1][j][1]);
                    dp[i][j][2]=max(dp[i-1][j-1][1]+1,dp[i-1][j][2]+1);
                }//与上面同理 
                ans=ans>dp[i][j][1]?ans:dp[i][j][1];
                ans=ans>dp[i][j][2]?ans:dp[i][j][2];
            }

代码（带调试）：

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<iostream>
using namespace std;
int dp[1000005][20][3];
char st[100005];
int main()
{
    int n,m,sum=0;
    scanf("%d%d\n",&n,&m);
    bool flag=true;
    bool ff=false;
    for(int i=1;i<=n;i++)
    {
        char ch;        
        scanf("%c",&ch);
        if(ch=='a') ff=true;
        if(ch=='b'&&flag)
            continue;
        flag=false;
        sum++;
        st[sum]=ch;
    }
    if(ff==false)
    {
        printf("0");
        return 0;
    }
    int ans=0;
    for(int i=1;i<=sum;i++)
    {
        for(int j=0;j<=m;j++)
        {
            cout<<endl;
            printf("st[%d]=%c  ",i,st[i]);
            if(j==0)
            {
                printf("j=0     ");
                if(st[i]=='b'){
                    dp[i][j][1]=dp[i-1][j][1];
                    printf("dp[%d][%d][1]=%d",i,j,dp[i][j][1]);
                }
                else{
                    dp[i][j][1]=dp[i-1][j][1]+1;
                    printf("dp[%d][%d][1]=%d  ",i,j,dp[i][j][1]);
                }
                ans=ans>dp[i][j][1]?ans:dp[i][j][1];
            }
            else
            {
                printf("j!!=0     ");
                if(st[i]=='a')
                {
                    dp[i][j][1]=max(dp[i-1][j-1][2]+1,dp[i-1][j][1]+1);
                    dp[i][j][2]=max(dp[i][j-1][1],dp[i-1][j][2]);
                    printf("dp[%d][%d][1]=%d  ",i,j,dp[i][j][1]);
                    printf("dp[%d][%d][2]=%d  ",i,j,dp[i][j][2]);
                }
                else
                {
                    dp[i][j][1]=max(dp[i-1][j-1][2],dp[i-1][j][1]);
                    dp[i][j][2]=max(dp[i-1][j-1][1]+1,dp[i-1][j][2]+1);
                    printf("dp[%d][%d][1]=%d  ",i,j,dp[i][j][1]);
                    printf("dp[%d][%d][2]=%d  ",i,j,dp[i][j][2]);
                }
                ans=ans>dp[i][j][1]?ans:dp[i][j][1];
                ans=ans>dp[i][j][2]?ans:dp[i][j][2];
            }
        }
    }
    printf("** ans=%d",ans);
    return 0;
}