删除子串
题目描述
给你一个长度为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,只改变了一次,且长度最长。
【题目分析】
这道题来自于牛客网的一场比赛,昨天第一次注册牛客网,打了一场比赛,然后就GG了,下来想了一下这道题,发现自己写得已经离AC很接近了,然后就把代码补上吧。
首先可以发现这道题其实算一道动规题,当然同机房的几位大佬又讲一些好似很厉害,很优秀的算法,不过这里就先写动规吧。。。
找状态转移方程是动规题的关键,由于这道题有“只由a、b组成的字符串”这条妙妙的性质,我们就可以利用这个妙妙的性质来写转移方程。
维护一个dp数组——>dp[i][j][k]。
[i]——>维护到了第几个字符
[j]——>表示我们使用了几次变化
[k]——>这一项只有两个,表示这次提取后的子串的末尾是a还是b。(这样就便于状态转移)
然后就可以写出状态转移方程
1 if(j==0) 2 { 3 if(st[i]=='b') 4 dp[i][j][1]=dp[i-1][j][1];//因为不用变化且字符串首项一定是a所以b不能被提取 5 else 6 dp[i][j][1]=dp[i-1][j][1]+1;//是a就可以提取了 7 ans=ans>dp[i][j][1]?ans:dp[i][j][1]; 8 } 9 else 10 { 11 if(st[i]=='a')//当前字符是a 12 { 13 dp[i][j][1]=max(dp[i-1][j-1][2]+1,dp[i-1][j][1]+1);//表示要选这个a 14 //选择的方式可以有两个,一个是用一次变化从b到a,或者不用变化从a到a 15 dp[i][j][2]=max(dp[i][j-1][1],dp[i-1][j][2]);//表示不选这个a 16 } 17 else 18 { 19 dp[i][j][1]=max(dp[i-1][j-1][2],dp[i-1][j][1]); 20 dp[i][j][2]=max(dp[i-1][j-1][1]+1,dp[i-1][j][2]+1); 21 }//与上面同理 22 ans=ans>dp[i][j][1]?ans:dp[i][j][1]; 23 ans=ans>dp[i][j][2]?ans:dp[i][j][2]; 24 }
写出了转移方程,那就很舒服了,在特判一下全为b的情况就OK了
【代码实现】
1 #include<cstdio>
2 #include<vector>
3 #include<queue>
4 #include<cstring>
5 using namespace std; 6 int dp[1000005][20][3]; 7 char st[100005]; 8 int main() 9 { 10 int n,m,sum=0; 11 scanf("%d%d\n",&n,&m); 12 bool flag=true; 13 bool ff=false; 14 for(int i=1;i<=n;i++) 15 { 16 char ch; 17 scanf("%c",&ch); 18 if(ch=='a') ff=true; 19 if(ch=='b'&&flag) 22 continue; 24 flag=false; 25 sum++; 26 st[sum]=ch; 27 } 28 if(ff==false) 29 { 30 printf("0"); 31 return 0; 32 } 33 int ans=0; 34 for(int i=1;i<=sum;i++) 35 { 36 for(int j=0;j<=m;j++) 37 { 38 if(j==0) 39 { 40 if(st[i]=='b') 41 dp[i][j][1]=dp[i-1][j][1]; 42 else
43 dp[i][j][1]=dp[i-1][j][1]+1; 44 ans=ans>dp[i][j][1]?ans:dp[i][j][1]; 45 } 46 else
47 { 48 if(st[i]=='a') 49 { 50 dp[i][j][1]=max(dp[i-1][j-1][2]+1,dp[i-1][j][1]+1); 51 dp[i][j][2]=max(dp[i][j-1][1],dp[i-1][j][2]); 52 } 53 else
54 { 55 dp[i][j][1]=max(dp[i-1][j-1][2],dp[i-1][j][1]); 56 dp[i][j][2]=max(dp[i-1][j-1][1]+1,dp[i-1][j][2]+1); 57 } 58 ans=ans>dp[i][j][1]?ans:dp[i][j][1]; 59 ans=ans>dp[i][j][2]?ans:dp[i][j][2]; 60 } 61 } 62 } 63 printf("%d",ans); 64 return 0; 65 }