hdu4333 Revolving Digits（扩展kmp+kmp最小循环节）

题目链接：点击打开链接

题意描述：给定一个字符串，可以把字符串的后x位移到字符串前面组成一个新的字符串？问对于所有的新组成的字符串中去掉重复的之后，比原串小的个数，等于的个数，大于的个数？

解题思路：扩展KMP（next1[i]数组含义：x[i···len-1]与x[0···len-1]的最长公共前缀）

分析：首先我们把字符串s复制一遍接到原字符串后面形成ss，这样在ss中以i（i>=0&&i<len）为起点的长度为len的字符串就是所有可能的新字符串；

讲到这里，问题转变为在ss中以i（i>=0&&i<len）为起点的长度为len的字符串与ss字符串前缀的关系了，明显的就是扩展KMP

if(next1[i]>=len) b++;//相等
else if(st[next1[i]]>st[i+next1[i]]) a++;///小于
else c++;///大于

注意一点，题目中要求是去掉重复的，刚开始没考虑内存，使用set判重，结果果断超内存，仔细分析会发现如果重复则原串s一定是循环串，使用kmp算法可以判定，并求出最小循环节，那么我们遍历的时候i的范围就变成最小循环节的大小了

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#define MAXN 200010
using namespace std;
int max(int a,int b)
{
    return a>b?a:b;
}
void getnext1(char* x,int m,int* next1)
{
    next1[0]=m;
    int j=0;
    while(j+1<m&&x[j]==x[j+1]) j++;
    next1[1]=j;
    int k=1;
    for(int i=2; i<m; i++)
    {
        int p=next1[k]+k-1;
        int L=next1[i-k];
        if(i+L<p+1) next1[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<m&&x[i+j]==x[j]) j++;
            next1[i]=j;
            k=i;
        }
    }
}
void getnext2(char* x,int m,int* next2)
{
    int i,j;
    j=next2[0]=-1;
    i=0;
    while(i<m)
    {
        while(-1!=j&&x[i]!=x[j]) j=next2[j];
        ++i,++j;
        next2[i]=j;
    }
}
char st[MAXN];
int next1[MAXN];
int next2[MAXN/2];
int main()
{
    int T;
    scanf("%d",&T);
    for(int t=1; t<=T; ++t)
    {
        scanf("%s",st);
        int len=strlen(st);

        getnext2(st,len,next2);
        int L;
        if(len%(len-next2[len])==0)
            L=len-next2[len];
        else
            L=len;

        for(int i=0; i<len; i++)
            st[i+len]=st[i];
        st[len<<1]='\0';
        getnext1(st,len*2,next1);

        int a,b,c;
        a=b=c=0;
        for(int i=0; i<L; i++)
        {
            if(next1[i]>=len) b++;
            else if(st[next1[i]]>st[i+next1[i]]) a++;
            else c++;
        }
        printf("Case %d: %d %d %d\n",t,a,b,c);
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。