牛客网暑期ACM多校训练营（第三场）E Sort String【KMP or hash】

题目链接：https://www.nowcoder.com/acm/contest/141/E

 

思路：

可以用KMP，也可以用hash。讲一下KMP的思路。

经过观察发现，如果字符串存在循环节，且长度为k，那么显然可以分k组。从一个循环节内不同位置开始，都可以分为不同组。那么问题就变成了找循环节。我们可以利用KMP的失配函数来寻找循环节。循环节的长度k=len-f[len]。如果len%k==0，则存在循环节；否则不存在。

参考文章：https://www.cnblogs.com/jackge/archive/2013/01/05/2846006.html

代码：

#include <stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAXN 1000050
#define ll long long
char P[MAXN];
int f[MAXN];
void getFail()
{
    int len=strlen(P);
    f[0]=f[1]=0;
    for(int i=1;i<len;i++)
    {
        int j=f[i];
        while(j && P[j]!=P[i]) j=f[j];
        f[i+1] = P[j]==P[i]? j+1:0;
    }
}

int main()
{
    scanf("%s",P);
    int len=strlen(P);
    getFail();
    int k=len-f[len];
    if(len%k)
    {
        printf("%d\n",len);
        for(int i=0;i<len;i++)
        {
            printf("1 %d\n",i);
        }
    }
    else
    {
        printf("%d\n",k);
        for(int i=0;i<k;i++)
        {
            printf("%d",len/k);
            for(int j=i;j<len;j+=k)
            {
                printf(" %d",j);
            }
            printf("\n");
        }
    }
    return 0;
}