每日算法题（Day3）----Power Strings

题目描述

原题：POJ 2406

给定若干个长度 ≤10^6的字符串，询问每个字符串最多是由多少个相同的子字符串重复连接而成的。如：ababab 则最多有 3 个 ab 连接而成。

输入格式

输入若干行，每行有一个字符串。特别的，字符串可能为 . 即一个半角句号，此时输入结束。

输出格式

输出若干行整数，表示最大字符串数。

分析

经典的KMP算法
随便找本数据结构的书就有相关内容，不再赘述

样例

abcd
aaaa
ababab
.
------------------------
1
4
3

代码

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
int nxt[1000100];
char s[1000100];
int n;
int main() {
    while (~scanf("%s", s + 1) && s[1] != '.') 
    {
        n = strlen(s + 1);
        nxt[1] = 0;
        for (int i = 2, j = 0; i <= n; i++) 
        {
            while (j > 0 && (j == n || s[i] != s[j + 1])) 
                j = nxt[j];
            if (s[i] == s[j + 1]) 
                j++;
            nxt[i] = j;
        }
        if (n % (n - nxt[n]) == 0 && nxt[n])
            printf("%d\n", n / (n - nxt[n]));
        else 
            puts("1");
    }
    return 0;
}