Wannafly挑戰賽12 C 刪除子串 三維DP||一維滾動 __Algorithm

連結：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行不用特判直接搞，簡直了。

My Code：

#include<iostream>#include<cstring>#include<algorithm>#include<iomanip>#include<cmath>using namespace std;const int maxn = 1e5 + 10;#define INF 0x3f3f3f3fchar s[maxn];int dp[maxn][12][2];//0為a，1為bint 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;}