hdu 4333(扩展KMP)

题意：给一个数字，每一次把它的最后一位拿到最前面，一直那样下去，分别求形成的数字小于，等于和大于原来数的个数。

例如：134可以形成134，341，413三个数，所以分别是1，1，1。

 

分析：

由于长度为len的字符串形成题目要求的串的个数为len，那么我们可以把原来的两个串T连接起来形成字符串S，然后找S的每

个后缀的前len个元素即可。

 

这里主要是如何比较的问题，对于字符串的比较，我们可以先求出他们的最长公共前缀长度，然后只需要比较一次就可以知道结果了。那么最长公共前缀怎么求，由于这里是一个串T与另一个串S，来求S的所有后缀与T的最长公共前缀长度，所以用扩展

KMP。如果extend[i]>=len，就说明与原来的相等了，否则如果S[i+extend[i]]<T[extend[i]]就说明小于，否则就是大

于。

PS：题意要求的是不重复的数字的个数，所以要将重复的去除。就是找到原来的字符串的最小循环节，然后所有的结果除一下循环次数就好了。找最小循环节的方法是对字符串进行KMP，然后len - next[len]就是最小循环节的长度。

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> P;
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define clr(x,y) memset(x,y,sizeof x)
#define PI acos(-1.0)
#define ITER set<int>::iterator
const int Mod = 1e9 + 7;
const int maxn = 2e5 + 10;
const int N = 2;

int nexts[maxn],extand[maxn];
char S[maxn],T[maxn];
int nextval[maxn];

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

void GetExtand(const char *S,const char *T) {
    Getnexts(T);
    int slen = strlen(S),tlen = strlen(T),a = 0;
    int MinLen = slen < tlen ? slen : tlen;
    while(a < MinLen && S[a] == T[a]) a++;
    extand[0] = a;
    a = 0;
    for(int k = 1; k < slen; k ++) {
        int p = a + extand[a] - 1, L = nexts[k - a];
        if( (k - 1) + L >= p) {
            int j = (p - k + 1) > 0 ? (p - k + 1) : 0;
            while(k + j < slen && j < tlen && S[k + j] == T[j]) j ++;
            extand[k] = j;
            a = k;
        } else
            extand[k] = L;
    }
}
void NextVal(char *T)
{
    int i=0,j=-1;
    nextval[0]=-1;
    int Tlen=strlen(T);
    while(i<Tlen)
    {
        if(j==-1||T[i]==T[j])
        {
            i++;
            j++;
            if(T[i]!=T[j]) nextval[i]=j;
            else  nextval[i]=nextval[j];
        }
        else j=nextval[j];
    }
}

int main()
{
    int Tcase;scanf("%d",&Tcase);
    for(int ii = 1; ii <= Tcase;ii ++)
    {
        scanf("%s",T);int len = strlen(T);strcpy(S,T);for(int i = len; i < len * 2;i ++)S[i] = T[i - len];S[len * 2] = '\0';
        GetExtand(S,T);int x = 0,y = 0,z = 0;NextVal(T);
        int temp = 1;if(len % (len - nextval[len]) == 0)temp = len / (len - nextval[len]);//len - nexts[len]为最小循环节
        for(int i = 0; i < len; i ++)
        {
            if(extand[i] >= len)y ++;
            else
            {
                if(S[i + extand[i]] > T[extand[i]])
                    z ++;
                else x ++;
            }
        }
        printf("Case %d: %d %d %d\n",ii,x/temp,y/temp,z/temp);
    }
    return 0;
}