HDU 4333 Revolving Digits

题意分析：

             给你一个数字，要求从右往左依次吧数字往前面放，求新的数字比原来数字小的个数，相等的个数，还有大于的个数。如果用暴力的话坑定会超时。如果熟悉扩展KMP算法的话可能会想到吧这个数字看成一个字符串S，然后把这个字符串扩展成2倍T，在依次求出扩展后的字符串的后缀依次与原来字符串前缀的公共长度。判断条件是当extend【i】大于等于原来字符串的长度len时，即extend[i]  >=len,说明新的数字等于原来数，如果T【i+extend[i]】> S【extend[i]】时，说明大于。或者小于。 还有一个容易忽略的问题是S可能有最小循环节，这样就可能的出来的数字重复，所以要去重。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define MA 1000000
using namespace std;
int next[MA],extend[MA],nextval[MA];

void getNext(char s[]) {
    int len=strlen(s),a=0;
    next[0]= len;
    while(a<len-1 && s[a] == s[a+1]) a++;
    next[1]  =a;
    a = 1;
    for(int k=2; k<len; k++) {
        int p=a+next[a] -1,L=next[k-a];
        if(k-1+L>=p) {
            int j=p+1-k>0? p+1-k: 0;
            while(k+j<len && s[k+j] == s[j]) j++;
            next[k]  = j;
            a = k;
        }
        else next[k]  = L;
    }
}

void getExtend(char s[],char T[]) {
    int len=strlen(s),lenp=strlen(T),a=0;
    int mi = min(len,lenp);
    while(a<mi && s[a] == T[a]) a++;
    extend[0] = a;
    a=0;
    getNext(T);
    for(int k=1; k<len; k++) {
        int p=a+extend[a] -1,L=next[k-a];
        if(k-1+L>=p) {
            int j=p+1-k>0? p+1-k : 0;
            while(j+k<len&&j<lenp &&s[j+k] == T[j]) j++;
            extend[k] = j;
            a = k;
        }
        else extend[k]  = L;
    }
}

void nextVal(char s[]) {
    int len=strlen(s),j=-1,i=0;
    nextval[0] = -1;
    while(i<len) {
        if(j==-1 || s[i] == s[j]) {
            ++i; ++j;
            if(s[i]!=s[j])
               nextval[i] = j;
            else nextval[i]  = nextval[j] ;
        }
        else j = nextval[j];
    }
}
int main() {
    int t;
    char s[MA],s1[MA];
    scanf("%d",&t);
    int ka=1;
    while(t--) {
        scanf("%s",s1);
        //cout<<t<<endl;
        strcpy(s,s1);
        strcat(s,s1);
        //cout<<s<<endl;
        nextVal(s1);
        int len=strlen(s1);
        int le=len-nextval[len],co=1;
        if(len%le==0) co = len/le;
        //cout<<next[len]<<endl;
        getExtend(s,s1);
        int su1=0,su2=0,su3=0;
        for(int i=0; i<len; i++) {
            if(extend[i] >=len) su1++;
            else if(s[i+extend[i]]>s1[extend[i]]) su2++;
            else su3++;
        }
        printf("Case %d: %d %d %d\n",ka++,su3/co,su1/co,su2/co);
    }
    return 0;
}