Wannafly挑战赛12 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 <= 10 ⁵,0 <= m <= 10）

第二行输入一行长度为n且由a和b组成的字符串

输出描述:

输出一个数字表示最长长度

DP

dp[i][mi][pre]表示遍历到第i个，变化为mi，且第i个字符前一个为pre字符时，已删除字符的最小值，然后用总长度减去最小值即可，我的方法讨论较多，较为复杂，且可用滚动数组化简(但我实在太懒...)，并且有大佬代码即简洁美观，又省空间省时间，在此也会贴上，只是我也不大理解...

本人代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int N = 100000 + 5;
int dp[N][15][2];
char s[N];
int min3(int a, int b, int c)
{
    return min(a, min(b, c));
}
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    scanf("%s", s);
    int pos;
    bool ju = true;
    for(int i = 0; s[i]; i++)
    {
        if(s[i] == 'a')
        {
            ju = false;
            pos = i;
            break;
        }
    }
    if(ju)
        printf("0\n");
    else
    {
        int posb = -1;
        for(int i = pos; i < n; i++)
        {
            if(s[i] == 'b')
            {
                posb = i;
                break;
            }
        }
        int maxi = n - pos;
        memset(dp, inf, sizeof(dp));
        for(int i = 0; i <= m; i++)
        {
            dp[pos][i][0] = 0;
        }
        for(int i = pos + 1; i < n; i++)
        {
            for(int j = 0; j <= m; j++)
            {
                int tp = s[i] - 'a';
                if(j != 0)
                    dp[i][j][tp] = min3(dp[i][j][tp], dp[i-1][j][tp], dp[i-1][j-1][tp^1]);
                else
                    dp[i][j][tp] = min(dp[i][j][tp], dp[i-1][j][tp]);
                if(!(tp == 0 && i < posb))
                {
                    if(j != 0)
                        dp[i][j][tp^1] = min3(dp[i][j][tp^1], dp[i-1][j][tp^1] + 1, dp[i-1][j-1][tp] + 1);
                    else dp[i][j][tp^1] = min(dp[i][j][tp^1], dp[i-1][j][tp^1] + 1);
                }
 
            }
 
        }
        int mini = inf;
        for(int i = 0; i <= m; i++)
        {
            mini = min3(mini, dp[n-1][i][0], dp[n-1][i][1]);
        }
        printf("%d\n", maxi - mini);
    }
}

大佬代码:

#include <bits/stdc++.h>
   
using namespace std;
#define mo 100
int dp1[mo],dp2[mo];
   
int main()
{
    long long n,m;
    while(cin>>n>>m)
    {
        string a;
        cin>>a;
   
        for(int i=0;a[i];i++)
        {
            for(int j=m;j>=1;j--)
            {
                if(a[i]=='a')
                {
                    dp1[j]=max(dp1[j]+1,dp2[j-1]+1);
                }
                else if(dp1[j-1])
                {
                    dp2[j]=max(dp2[j]+1,dp1[j-1]+1);
                }
            }
            if(a[i]=='a') dp1[0]++;
   
             
        }
        cout<<max(dp1[m],dp2[m])<<endl;
   
   
    }
}