hdu4333 扩展kmp

#include <iostream>
#include <algorithm>
#include <string.h>
#include <cstdio>
using namespace std;
const int MM=2000005;
int extand[MM],f[MM];
char S[MM];
void GetNext(const char *T){
     int len=strlen(T),a=0;
     extand[0]=len;
     while(a<len-1 && T[a]==T[a+1]) a++;
     extand[1]=a;
     a=1;
     for(int k=2;k<len;k++){
         int p=a+extand[a]-1,L=extand[k-a];
         if( (k-1)+L >= p){
             int j = (p-k+1)>0 ? (p-k+1) : 0;
             while(k+j<len && T[k+j]==T[j]) j++;
             extand[k]=j;
             a=k;
         }
         else
             extand[k]=L;
     }
}
void getFail(char *P,int *f)
{
    int m=strlen(P);
    f[0]=0;
    f[1]=0;
    for(int i=1; i<m; i++){
        int j=f[i];
        while(j&&P[i]!=P[j])j=f[j];
        f[i+1]=P[i]==P[j]?j+1:0;
    }

}
int main(){
    int cas;
    scanf("%d",&cas);
    for(int cc=1; cc<=cas; cc++)
    {
          scanf("%s",S);
          int len=strlen(S);
          for(int i=0; i<len; i++)
            S[len+i]=S[i];
          S[len+len]=0;
          GetNext(S);
          int l=0,e=0,g=0;
          for(int i=0; i<len; i++)
            if(extand[i]>=len)e++;
            else {
                if(S[extand[i]+i]>S[extand[i]]){
                    g++;
                }else l++;

            }
            getFail(S,f);
        int tem=1;
        if( ( len%(len-f[len]) )==0  )tem=len/(len-f[len]);
        printf("Case %d: %d %d %d\n",cc,l/tem,e/tem,g/tem);
    }
    return 0;
}