Revolving Digits(kmp与扩展kmp)(解释)

原题地址

题意：

给你一个整数n，每次将n的最后一位放到第一位形成一个新数，这个操作重复n的位数次，求形成的所以数中，比原数n小、相等、大的各有几个。

思路：

理解花了点时间，orz，错的地方欢迎指出
看一下数据范围10^100000，50组输入，基本就告别暴力了
然后不暴力怎么做呢。。。肯定是字符串。
我们如果将操作的过程反转，即将第一位放到最后一位去，就变成了n的后缀。
然后扩展kmp的extand[i]数组就表示原串的后缀与模式串的最长公共前缀
设原串长度为l，原串为ch。
如果我们将原串复制再链接到原串后面(即123变成123123)，则对于原串的每个后缀，如果我们只取其l位，则这l位就是我们要求的一个新数。
因为比较大小只需要比较从前往后第一个不相等的数的大小就可以了；
即比较ch[extand[i]]和ch[i+extand[i]]的大小就可以。
然后我们再考虑如原串为1111或1212这种情况，它们都出现了循环，所以我们需要找出并去掉循环节。
循环的原因是因为原串的前缀和真后缀出现了相同的情况而导致的循环，所以我们需要找出其前缀和真后缀的最长公共前缀长度。也就是kmp的next[]数组
然后就是考虑诸如12312这种虽然next[l]不等于0，但并没有出现循环的情况，
所以只有l%(l-next[l])==0才会导致循环节的出现，并且每个数一定都出现了l/(l-next[l])次，所以我们只需要将得出的结果除以出现的次数就可以了。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <vector>
#include <cmath>
#include <queue>
#include <set>
#include <map> 
#include <algorithm>
using namespace std;
char ch[200000];
int next1[200000],l,extand[200000];
void getnext(int n)
{
	int i=0,j=-1;
	next1[0]=-1;
	while(i<n)
	{
		while(j!=-1&&ch[i]!=ch[j])
            j=next1[j];
        if(ch[++i]==ch[++j])
           next1[i]=next1[j];
        else
           next1[i]=j;
	}
}
void getex(int n){
     int a=0;
     extand[0]=n;
     while(a<n-1&&ch[a]==ch[a+1]) 
	    a++;
     extand[1]=a;
     a=1;
     for(int k=2;k<n;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<n&&ch[k+j]==ch[j]) 
			     j++;
             extand[k]=j;
             a=k;
         }
         else extand[k]=L;
     }
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int k=1;k<=n;k++){
		memset(next1,0,sizeof(next1));
		memset(extand,0,sizeof(extand));
		scanf("%s",ch);
		l=strlen(ch);
        getnext(l);
        int tt;   //求循环节 
        if(l%(l-next1[l])==0)  
		  tt=l/(l-next1[l]); 
		else  
		  tt=1;
		for(int i=0;i<l;i++)
		  ch[l+i]=ch[i];
		getex(2*l);   //两倍是为了方便求原串的各个后缀 
		int L=0,E=0,G=0;
		for(int i=0;i<l;i++){
			if(extand[i]>=l)   E++;   
			else if(ch[extand[i]]>ch[i+extand[i]])  L++;   //判断大小 
			else G++;
		}
		printf("Case %d: %d %d %d\n",k,L/tt,E/tt,G/tt);   //去掉循环节 
	}
	return 0;
}