一道双指针的题(音乐家的曲调 )

• 题目链接： https://ac.nowcoder.com/acm/problem/222216

• 题意：
  从长度为n的只有小写字母的字符串中选出三个不相交的区间，使这三个区间的区间长度之和最大。当然，要求选出的任意一个区间，区间内每种小写字母的出现次数不能超过m。
  请问最大的区间长度之和是多少?

• 思路：
  数据范围给了1e7 ，只能O(n) 求解
  思考如何在O(n)中得到三个不相交区间的最大总长度
  我们可以通过双指针，
  预处理出来从1~i的这个范围内的最大前缀 以及 i~n 范围内的最大后缀区间长度
  最后再通过双指针拼接一下， 总的3个区间长度就是 从左往右扫，
  ans[i] = max(ans[i-1],i-j+1 + pre[j-1] + back[i+1]);
  具体看代码

#include <bits/stdc++.h>
using namespace std;
#define rep(x, y, z) for (int x = y; x <= z; x++)
#define dec(x, y, z) for (int x = y; x >= z; x--)
const int N = 1e7+5, mod = 1e9 + 7;
const double expp = 1e-10;
typedef long long ll;
//------------------------
int n,m;
char s[N];
int pre[N];
int back[N];
int cnt[30];
int ans[N];
int main(){
    scanf("%d%d%s",&n,&m,s+1);

    for(int i=1,j=1;i<=n;i++){
        cnt[s[i]-'a'] ++;
        while(j<=i && cnt[s[i]-'a']>m){
              cnt[s[j]-'a'] --;
              j++;
        }
        
        pre[i] = max(pre[i-1],i-j+1);
    }
    memset(cnt,0,sizeof cnt);

    for(int i=n,j=n;i;i--){
        cnt[s[i]-'a'] ++;
        while(j>=i && cnt[s[i]-'a']>m){
              cnt[s[j]-'a'] --;
              j--;
        }
        
        back[i] = max(back[i+1],j-i+1);
    }
    memset(cnt,0,sizeof cnt);

    for(int i=1,j=1;i<=n;i++){
        cnt[s[i]-'a'] ++;
        while(j<=i && cnt[s[i]-'a']>m){
              cnt[s[j]-'a'] --;
              j++;
        }

        //  pre[i] = max(pre[i-1],i-j+1);
        ans[i] = max(ans[i-1],i-j+1 + pre[j-1] + back[i+1]);
    }
    int res = 0;
    rep(i,1,n) res = max(res,ans[i]);
    cout<<res<<endl;




    return 0;
}