Wannafly挑战赛12 C 删除子串【三维DP||一维滚动】

链接：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;
}