Newcoder 141 E.Sort String（kmp）

Description

给出一个只由小写字母组成的字符串$S$，将该字符串旋转平移生成$n$个字符串，对于相同的字符串，将其在原串中的起始位置分到一个集合，把每个集合元素从小到大排序，再把每个集合按元素最小值从小到大排序，输出排序后的这些集合以及每个集合内的元素

Input

一个只由小写字母组成的字符串$S(1\le |S|\le 10^6)$

Output

输出集合个数以及每个集合

Sample Input

abab

Sample Output

2
2 0 2
2 1 3

Solution

若存在某个集合里元素超过一个，说明该串必然有若干循环节组成，否则该串生成的所有串都不相同，用$kmp$跑一遍该串，$x=n-next[n]$即为该串的循环节长度，但是注意只有当$x|n$时该串才由若干循环节组成，此处每组有$\frac{n}{x}$个元素，第$i$组第$j$个元素起点即为$j\cdot x+i,0\le i<x,0\le j<\frac{n}{x}$

Code

#include<cstdio>
#include<cstring>
using namespace std;
#define maxn 1000005
char s[maxn];
int nex[maxn];
int main()
{
    scanf("%s",s);
    int len=strlen(s),i,j;
    for(i=0,j=-1;i<=len;j++,i++)
    {
        nex[i]=j;
        while(j>-1&&s[i]!=s[j])j=nex[j];
    }
    int x=len-nex[len];
    if(len%x!=0)
    {
        printf("%d\n",len);
        for(int i=0;i<len;i++)printf("1 %d\n",i);
    } 
    else
    {
        printf("%d\n",x);
        for(int i=0;i<x;i++)
        {
            printf("%d",len/x);
            for(int j=0;j<len/x;j++)printf(" %d",j*x+i);
            printf("\n");
        }
    }
    return 0;
}