链接:https://www.nowcoder.com/acm/contest/79/C
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
给你一个长度为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,只改变了一次,且长度最长。
dp[i][j][k]表示第0~i的字符串,变数维j,以k结尾的最长长度。预处理一下,三层循环便可以解决问题。一开始知道可以用滚动数组,没用,昨晚才发现自己和滚动大佬的差距还是那么的大。
我的方法,算是常规思路吧,各种卡细节,四五个特判。大佬的代码26行不用特判直接搞,简直了!
我的代码:
#include<iostream>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
using namespace std;
const int maxn = 1e5 + 10;
#define INF 0x3f3f3f3f
char s[maxn];
int dp[maxn][12][2];//0为a,1为b
int main()
{
int n, m;
scanf("%d%d", &n, &m);
scanf("%s", s);
for (int i = 0; i < maxn; i++) {
for (int j = 0; j < 12; j++) {
dp[i][j][0] = dp[i][j][1] = -INF;
}
}
int tmp = n;
if (s[0] == 'a') {
for (int i = 0; i < n; i++) {
if (s[i] != 'a') {
tmp = i; dp[tmp - 1][0][0] = tmp;
dp[tmp][1][1] = tmp + 1;
//cout << dp[tmp - 1][0][0] << endl;
//cout << dp[tmp][1][1] << endl;
break;
}
}
if (tmp == n) {
cout << n << endl;
return 0;
}
}
else
{
for (int i = 0; i < n; i++) {
if (s[i] != 'b') { tmp = i; break; }
}
if (tmp == n) {
cout << "0" << endl;
return 0;
}
int stmp = 0;
for (int i = tmp; i < n; i++) {
if (s[i] != 'a') {
stmp = i; break;
}
}
if (stmp > tmp) {
dp[stmp][0][0] = stmp - tmp;
dp[stmp][1][1] = stmp - tmp + 1;
tmp = stmp;
}
else {
cout << n - tmp << endl;
return 0;
}
/*cout << dp[stmp][0][0] << endl;
cout << dp[stmp][1][1] << endl;*/
}
for (int i = 1; i < n; i++) {
dp[i][0][0] = max(dp[i][0][0], dp[i - 1][0][0]);
if (s[i] == 'a')
dp[i][0][0] = max(dp[i][0][0],dp[i - 1][0][0]+1);
for (int j = 1; j <= m; j++) {
dp[i][j][1] = max(dp[i - 1][j][1], dp[i][j][1]);
dp[i][j][0] = max(dp[i - 1][j][0], dp[i][j][0]);
if (s[i] == 'a') {
dp[i][j][0] = max(max(dp[i-1][j][0]+1,dp[i-1][j-1][1]+1),dp[i][j][0]);
//dp[i][j][1] = dp[i-1][j][1];
//cout << dp[i][j][0] << endl;
//cout << dp[i][j][1] << endl;
}
else {
dp[i][j][1] = max(max(dp[i-1][j][1]+1,dp[i-1][j-1][0]+1),dp[i][j][1]);
//cout << dp[i][j][1] << endl;
}
/*cout << dp[i][j][0] << endl;
cout << dp[i][j][1] << endl;*/
}
}
int ans = -INF;
for (int i = 0; i <= m; i++) {
ans = max(ans, dp[n-1][i][0]);
ans = max(ans, dp[n-1][i][1]);
}
cout << ans << endl;
return 0;
}
大佬的代码:
#include<cstdio>
#include<algorithm>
using namespace std;
char s[100100];
int a[13],b[13];
int main()
{
int N,M;
scanf("%d%d%s",&N,&M,s);
int k=0;
while(s[k]=='b') k++;
for(;k<N;k++)
{
if(s[k]=='a') a[0]++;
for(int i=1;i<=M;i++)
if(s[k]=='a')
a[i]=max(a[i],b[i-1])+1;
else
b[i]=max(b[i],a[i-1])+1;
}
int ans=0;
for(int i=0;i<=M;i++)
ans=max(ans,max(a[i],b[i]));
printf("%d\n",ans);
return 0;
}