hdu 4333 Revolving Digits(扩展KMP)

设主串长为len，将主串扩展为2倍长度主串，如“341”变为“341341”，在遍历中遍历len长度即可得到全部可能生成的串，如“341、413、134”。对于重复串问题，只有这个串具有循环节的时候才会产生重复串，用KMP的next数组求出最小循环节，用长度除以最小循环节得到循环节个数，再将最后答案除以循环节个数即可。

#include <iostream>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <stdio.h>
using namespace std;
const int MAXN=2000010;

int next1[MAXN];
int extend[MAXN];

void EKMP(char s[],char t[])//s为主串，t为模板串
{
    int i,j,p,L;
    int lens=strlen(s);
    int lent=strlen(t);
    next1[0]=lent;
    j=0;
    while(j+1<lent && t[j]==t[j+1])j++;
    next1[1]=j;

    int a=1;
    for(i=2;i<lent;i++)
    {
        p=next1[a]+a-1;
        L=next1[i-a];
        if(i+L<p+1)next1[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lent && t[i+j]==t[j])j++;
            next1[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=next1[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;
        }
    }
}

void getNext(char T[],int len)
{
    int j,k;
    j=0;k=-1;next1[0]=-1;
    while(j<len)
    {
        if(k==-1 || T[j]==T[k])
            next1[++j]=++k;
        else k=next1[k];
    }
}
char str1[MAXN],str2[MAXN];
int main()
{
    int T;
    int iCase=0;
    scanf("%d",&T);
    while(T--)
    {
        iCase++;
        scanf("%s",str1);
        int len=strlen(str1);
        strcpy(str2,str1);
        strcat(str2,str1);
        EKMP(str2,str1);
        int cnt1=0,cnt2=0,cnt3=0;

        for(int i=0;i<len;i++)//str2主串 str1模式串
        {
            if(extend[i]>=len)cnt2++;
            else
            {
                if(str2[i+extend[i]]<str1[extend[i]])cnt1++;
                else cnt3++;
            }
        }
        getNext(str1,len);
        int t=len-next1[len];//循环节长度
        int tol=1;
        if(len%t==0)tol=len/t;
        printf("Case %d: %d %d %d\n",iCase,cnt1/tol,cnt2/tol,cnt3/tol);
    }
    return 0;
}