HDU-4333 Revolving Digits

题目链接：https://vjudge.net/problem/HDU-4333

把数字当做字符串来处理，将字符串复制一遍末尾，再用扩展KMP求出最长公共前缀

接下来只需要让s[next[i]]与s[i+next[i]]比较就可以判断哪个串大

注意要先用KMP算一遍next[]求出循环节，计算的数字会有重复的，除以循环节才是正确答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e5+10;
char s[N*2];
int next0[N*2];
void get_next(char *a,int len){
	int i=-1,j=0;
	next0[0]=-1;
	while(j<len){
		if(i == -1 || a[i] == a[j])next0[++j]=++i;
		else i=next0[i];
	}
}
void get_extend(char *a,int len){
	int k=0,i=1;
	next0[0]=len;
	while(k+1<len && a[k] == a[k+1])++k;
	next0[1]=k;
	k=1;
	while(++i<len/2){//只需要求到原串的长度即可 
		int maxr=k+next0[k]-1;
		next0[i]=min(next0[i-k],max(maxr-i+1,0));
		while(i+next0[i]<len && a[next0[i]] == a[i+next0[i]])++next0[i];
		if(i+next0[i]>k+next0[k])k=i;
	}
}
int main()
{
	int T,kase=0;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s",s);
		int len=strlen(s);
		get_next(s,len); //求KMP的next数组 
		int k=len-next0[len]; 
		if(len%k==0) k=len/k; //可以用KMP求出字符串循环节 k
		else k=1;
		for(int i=0;i<len;i++) s[i+len]=s[i];
		get_extend(s,len+len);
		int a=0,b=0,c=0;
		for(int i=0;i<len;i++)
		{
			if(next0[i]>=len) b++;
			else if(s[i+next0[i]]<s[next0[i]]) a++;
			else c++;
		}
		printf("Case %d: %d %d %d\n",++kase,a/k,b/k,c/k); //只记录不同的数，除以 k 
	}
	return 0;
}