poj 2406 求字符串中重复子串的个数

Sample Input

abcd
aaaa
ababab
.
Sample Output

1 //1个abcd
4 //4个a
3 //3个ab

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;

const int MAXN=1000010;
char T[MAXN];
int next[MAXN];
int n;
void getNext()
{
int j,k;
j=0;
k=-1;
next[0]=-1;
while(j < n)
{
if(k==-1||T[j]==T[k])
{
j++;
k++;

next[j]=k;
}
else k=next[k];

}
if(n%(n-k)==0)
printf("%d\n",n/(n-k));
else
printf("1\n") ;
}
int main()
{

while(scanf("%s",&T) != EOF)
{
if (T[0]=='.')
break ;
n = strlen(T) ;
getNext();

}
return 0;
}