|poj 2406|KMP|Power Strings

poj传送门
可以知道，在一个字符串里的最短周期是$ms=n-f[i]$, 其中$f$是kmp中的失配函数。
如果$ms|n$, 那么输出$n/ms$, 否则输出 1 <script type="math/tex" id="MathJax-Element-45">1</script>.

#include<cstdio>    
#include<algorithm>    
#include<cstring>     
#define ms(i,j) memset(i,j, sizeof i);    
using namespace std;
const int MAXN = 1000000 + 5;
int n;
char s[MAXN];
int f[MAXN];
int main()   
{     
    while (scanf("%s", s)&&(s[0]!='.'))
    {
        n = strlen(s);
        f[0] = f[1] = 0;
        for (int i=1;i<n;i++)
        {
            int j = f[i];
            while (j && s[i]!=s[j]) j = f[j];
            f[i+1] = (s[i]==s[j]) ? (j+1) : (0);
        }
        int ans = n-f[n];
        if (n%ans==0) ans = n/ans; else ans = 1;
        printf("%d\n", ans);
    }
    return 0;    
}