HDU - 4333 Revolving Digits（扩展KMP）

http://acm.hdu.edu.cn/showproblem.php?pid=4333

题意

一个数字，依次将第一位放到最后一位，问小于本身的数的个数及等于本身的个数和大于本身的个数，但是要注意重复的不再计算

分析

当这个串有循环节时才会出现重复串，用KMP的next数组来计算循环节：len-next[len]。最后将得到的答案除以循环节个数即可。然后用扩展KMP，求出主串的后缀与模式串的前缀的最长公共前缀，这里的模式串是原串，而主串则是将两个原串连接起来。利用extend数组，当extend[i]>=n时，说明与原串匹配。否则则比较S[extend[i]]和S[i+extend[i]]的大小（比较不匹配的第一个位置）。

#include<iostream>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<cstdio>
#include<algorithm>
#include<map>
#include<set>
#define rep(i,e) for(int i=0;i<(e);i++)
#define rep1(i,e) for(int i=1;i<=(e);i++)
#define repx(i,x,e) for(int i=(x);i<=(e);i++)
#define X first
#define Y second
#define PB push_back
#define MP make_pair
#define mset(var,val) memset(var,val,sizeof(var))
#define scd(a) scanf("%d",&a)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define scddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define pd(a) printf("%d\n",a)
#define scl(a) scanf("%lld",&a)
#define scll(a,b) scanf("%lld%lld",&a,&b)
#define sclll(a,b,c) scanf("%lld%lld%lld",&a,&b,&c)
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#define lc idx<<1
#define rc idx<<1|1
#define lson l,mid,lc
#define rson mid+1,r,rc
using namespace std;
typedef long long ll;
template <class T>
void test(T a){cout<<a<<endl;}
template <class T,class T2>
void test(T a,T2 b){cout<<a<<" "<<b<<endl;}
template <class T,class T2,class T3>
void test(T a,T2 b,T3 c){cout<<a<<" "<<b<<" "<<c<<endl;}
const int N = 1e6+;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3fll;
const ll mod = ;
int T;
void testcase(){
    printf("Case %d:",++T);
}
const int MAXN = 2e5+;
const int MAXM = ;
const int p = ;

int nxt[MAXN],extend[MAXN];
char s[MAXN],t[MAXN>>];

void get_next(int m){
    int i,j;
    j=nxt[]=-;
    i=;
    while(i<m){
        while(-!=j&&t[i]!=t[j]) j=nxt[j];
        if(t[++i]==t[++j]) nxt[i]=nxt[j];
        else nxt[i]=j;
    }
}

void pre_EKMP(int m){
    nxt[]=m;
    int j=;
    while(j+<m&&t[j]==t[j+]) j++;
    nxt[]=j;
    int k=;
    for(int i=;i<m;i++){
        int p =nxt[k]+k-;
        int L=nxt[i-k];
        if(i+L<p+) nxt[i]=L;
        else{
            j=max(,p-i+);
            while(i+j<m&&t[i+j]==t[j]) j++;
            nxt[i]=j;
            k=i;
        }
    }
}

void EKMP(int m,int n){
    pre_EKMP(m);
    int j=;
    while(j<n&&j<m&&t[j]==s[j]) j++;
    extend[]=j;
    int k=;
    for(int i=;i<n;i++){
        int p =extend[k]+k-;
        int L=nxt[i-k];
        if(i+L<p+) extend[i]=L;
        else{
            j=max(,p-i+);
            while(i+j<n&&j<m&&s[i+j]==t[j]) j++;
            extend[i]=j;
            k=i;
        }
    }
}

void solve(){
    int lt=strlen(t);
    int ls=strlen(s);
    get_next(lt);
    int k=lt-nxt[lt];
    k= lt%k==?(lt/k):;
    EKMP(lt,ls);
    int e=,g=,l=;
    for(int i=;i<lt;i++){
        if(extend[i]>=lt) e++;
        else if(s[extend[i]]>s[i+extend[i]]) l++;
        else g++;
    }
    printf(" %d %d %d\n",l/k,e/k,g/k);
}
int main() {
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif // LOCAL
    int tt;
    T=;
    scd(tt);
    while(tt--){
        scanf("%s",s);
        strcpy(t,s);
        strcat(s,t);
        testcase();
        solve();
    }
    return ;
}