HDU4333 Revolving Digits（KMP+扩展KMP）

题意：给出一个十万位以内的数字，不断将最后的数字添加到前面去，如123就可以得到312、231、123.问最终得到的有多少个数字小于原数、等于原数、大于原数。

思路：十万位的长度，显然要用字符串来存，因为需要循环移位比较，所以直接将原串在末尾复制一遍。发现可以以复制后得到的s1为目标串，原串s2为模式串，做一次扩展KMP，每次只需要比较匹配长度的后面一位即可，这样就可以接近线性复杂度求解。

没有考虑去重WA了一次。易证，如果出现重复的情况那么一定是原串内存在循环。这时候需要用到KMP算法的思想求出循环节长度。求解的时候只需要跑一个循环节的长度就够了，剩下的都是重复结果。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
const int mod = 1000000007;
const int maxn=100005;
typedef long long ll;
int nxt[maxn];
//获得子串的next数组
void get_next(char str[]){
    int i = 0, j, pos;
    int len = strlen(str);
    nxt[0] = len;
    while(i + 1 < len && str[i] == str[i + 1]){
        ++i;
    }
    nxt[1] = i;
    pos = 1;
    for(i=2; i<len; ++i){
        if(nxt[i - pos] < pos + nxt[pos] - i){
            nxt[i] = nxt[i - pos];
        } else {
            j = nxt[pos] + pos - i;
            if(j < 0){
                j = 0;
            }
            while(i + j < len && str[i + j] == str[j]){
                ++j;
            }
            nxt[i] = j;
            pos = i;
        }
    }
}
//计算extend数组
void EXKMP(char target[], char pattern[], int extend[]){
    int i = 0, j, pos;
    get_next(pattern);
    int len1 = strlen(target);
    int len2 = strlen(pattern);
    while(i < len1 && i < len2 && target[i] == pattern[i]){
        ++i;
    }
    extend[0] = i;
    pos = 0;
    for(i=1; i<len1; ++i){
        if(nxt[i - pos] < pos + extend[pos] - i){
            extend[i] = nxt[i - pos];
        } else {
            j = extend[pos] + pos - i;
            if(j < 0){
                j = 0;//从头匹配
            }
            while(i + j < len1 && j < len2 && target[i + j] == pattern[j]){
                ++j;
            }
            extend[i] = j;
            pos = i;//更新pos
        }
    }
}
void get_next(char s[], int next[]){//KMP算法中的获取next数组
    int len = strlen(s);
    next[0] = -1;
    int index;
    for(int i=1; i<len; ++i){
        index = next[i - 1];
        while(index >= 0 && s[i] != s[index + 1]){
            index = next[index];
        }
        if(s[i] == s[index + 1]){
            next[i] = index + 1;
        } else {
            next[i] = -1;
        }
    }
}
char s1[maxn * 2], s2[maxn];
int extend[maxn * 2];
int main(){
    int t, cas = 0;
    scanf("%d", &t);
    while(t--){
        scanf("%s", s1);
        int len = strlen(s1);
        for(int i=0; i<len; ++i){
            s1[i + len] = s2[i] = s1[i];
        }
        s1[len + len] = s2[len] = 0;
        get_next(s2, nxt);
        if(nxt[len - 1] != -1 && len % (len - 1 - nxt[len - 1]) == 0){
            len = len - 1 - nxt[len - 1];
        }//如果存在循环节，就把len更新为循环节长度
        EXKMP(s1, s2, extend);
        int l = 0, e = 0, g = 0;
        for(int i=0; i<len; ++i){
            if(extend[i] >= len){
                ++e;
            } else if(s1[i + extend[i]] > s2[extend[i]]){//只需要比较匹配长度的后面一位即可
                ++g;
            } else {
                ++l;
            }
        }
        printf("Case %d: %d %d %d\n", ++cas, l, e, g);
    }
    return 0;
}