HDU 4333 Revolving Digits-----拓展KMP

此题就是比较裸的拓展KMP。

此SoL【转自KuangBin】：扩展KMP能求出一个串所有后缀串(即s[i...len])和模式串的最长公共前缀。于是只要将这个串复制一遍，求出该串每个后缀与其本身的最长公共前缀即可，当公共前缀>=len时，显然相等，否则只要比较下一位就能确定这个串与原串的大小关系。

　　至于重复串的问题，只有当这个串有循环节的时候才会产生重复串，用KMP的next数组求出最小循环节，用长度除以最小循环节得到循环节个数，在将3个答案都除以循环节个数即可。

这题搞出循环节就ok了。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;

const int maxn = 2000000 + 10;

int next[maxn],extend[maxn];

//s主串  t子串 
inline void ekmp(char *s,char *t)
{
    int i,j,p,L;
    int lens=strlen(s);
    int lent=strlen(t);
    next[0]=lent;
    j=0;
    while(j+1<lent&&t[j]==t[j+1])  j++;
    next[1]=j;
    int a=1;
    for(i=2;i<lent;i++)
    {
        p=next[a]+a-1;
        L=next[i-a];
        if(i+L<p+1)  next[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lent&&t[i+j]==t[j])  j++;
            next[i]=j;
            a=i;
        }
    }
    j=0;
    while(j<lens&&j<lent&&s[j]==t[j])  j++;
    extend[0]=j;
    a=0;
    for(i=1;i<lens;i++)
    {
        p=extend[a]+a-1;
        L=next[i-a];
        if(L+i<p+1)  extend[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lens&&j<lent&&s[i+j]==t[j])  j++;
            extend[i]=j;
            a=i;
        }
    }
}

inline void get_next(char *T,int len)
{
    int j,k;
    j=0;k=-1;next[0]=-1;
    while(j<len)
    {
        if(k==-1||T[j]==T[k])
            next[++j]=++k;
        else k=next[k];
    }
}

char str1[maxn],str2[maxn];

int main()
{
    int T,text=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",str1);
        int len=strlen(str1);
        strcpy(str2,str1);
        strcat(str2,str1);
        ekmp(str2,str1);
        int cnt1,cnt2,cnt3;
        cnt1=cnt2=cnt3=0;
        for(int i=0;i<len;i++)
        {
            if(extend[i]>=len)  cnt2++;
            else
            {
                if(str2[i+extend[i]]<str1[extend[i]])  cnt1++;
                else cnt3++;
            }
        }
        get_next(str1,len);
        int t=len-next[len];
        int total=1;
        if(len%t==0)  total=len/t;
        printf("Case %d: %d %d %d\n",text++,cnt1/total,cnt2/total,cnt3/total);
    }
    return 0;
}