【HDU】3474 Necklace

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 2000010
char s[MAXN];
int n,len,q[MAXN],sum[MAXN];
bool ok[2][MAXN];
void Read()
{
    int i;
    scanf(" %s",s+1);
    len=strlen(s+1);
    for(i=1;i<=len;i++)
        s[i+len]=s[i];
    n=len<<1;
}
void DoIt(int k)
{
    int i,front,rear;
    sum[0]=0;
    for(i=1;i<=n;i++)
    {
        if(s[i]=='C')
            sum[i]=sum[i-1]+1;
        else
            sum[i]=sum[i-1]-1;
    }
    front=0;
    rear=-1;
    for(i=1;i<len;i++)
    {
        for(;front<=rear&&sum[q[rear]]>=sum[i];rear--);
        q[++rear]=i;
    }
    for(;i<=n;i++)
    {
        for(;front<=rear&&i-q[front]>=len;front++);
        for(;front<=rear&&sum[q[rear]]>=sum[i];rear--);
        q[++rear]=i;
        ok[k][i-len]=(sum[q[front]]>=sum[i-len]);
    }
}
int main()
{
    int c,i,cnt,ca=1;
    scanf("%d",&c);
    while(c--)
    {
        Read();
        DoIt(0);
        reverse(s+1,s+1+n);
        DoIt(1);
        for(i=cnt=0;i<len;i++)
        {
            if(ok[0][i]||ok[1][len-i])
                cnt++;
        }
        printf("Case %d: %d\n",ca++,cnt);
    }
    return 0;
}