牛客练习赛85 B 音乐家的曲调（尺取法）

Posted 2021-07-31 CCSU_Cola

题目链接

题意：在字符串中寻找三个字串，使得它们三个串的长度最大，每个串需要满足同一字符出现的次数不大于m次。

思路：简单尺取法选取的是单一长度最大，且同一字符出现次数不大于m次。由于需要选取三段，我们可以先预处理一个前缀满足条件的最大长度以及后缀满足条件的最大的长度，然后最后再跑一次整个串，取一个max(L[l-1]+R[r+1]+(r-l+1)即可。

代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
char p[10000010];
int lll[10000010],rrr[10000010];
int vis[200];
int idx=3;
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    scanf("%s",p+1);
    int l=1,r=1;
    while(r<=n&&l<=r){
        vis[p[r]-'a']++;
        while(vis[p[r]-'a']>m){
            vis[p[l]-'a']--;
            l++;
        }
        lll[r]=max(lll[r-1],(r-l)+1);
        r++;
    }
    l=n,r=n;
    memset(vis,0,sizeof vis);
    while(r>=l&&l>=1){
        vis[p[l]-'a']++;
        while(vis[p[l]-'a']>m){
            vis[p[r]-'a']--;
            r--;
        }
        rrr[l]=max(rrr[l+1],(r-l)+1);
        l--;
    }
    l=1,r=1;
    int ans=0;
    memset(vis,0,sizeof vis);
    while(r<=n&&l<=r){
        vis[p[r]-'a']++;
        while(vis[p[r]-'a']>m){
            vis[p[l]-'a']--;
            l++;
        }
        ans=max(ans,lll[l-1]+rrr[r+1]+(r-l)+1);
        r++;
    }
    printf("%d\\n",ans);
}