E. Eggfruit Cake（尺取）

原题

题意

就是说给一个字符串和一个数字n，求这个字符串环成环的所含有的长度小于等于S且至少含有一个E的子串有多少个。

代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#define LL long long
#define INF 0x3f3f3f3f
const int MAX=1e6+10;
using namespace std;

char B[MAX];
int num[MAX];
int main()
{
    int cnt=0,m,n;
    cin>>B+1>>m;
    n=strlen(B+1);
    for(int i=1;i<=n;i++)
        B[i+n]=B[i];
    for(int i=0;i<=2*n;i++)
        if(B[i]=='E')
            num[++cnt]=i;
    if(!cnt)
    {
        printf("0\n");
        return 0;
    }
    int last=1;
    LL ans=0;
    for(int i=1;i<=n;i++)
    {
        if(num[last]<i&&last<=cnt)
            last++;
        if(last>cnt)
            break;
        ans+=max(0,i+m-num[last]);
    }
    printf("%lld\n",ans);
    return 0;
}