hdu 4333(扩展KMP)

来源:互联网 发布:javascript代码例子 编辑:程序博客网 时间:2024/06/13 02:03


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

例如: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>::iteratorconst 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;}