POJ-2406-Power Strings

ACM模版

描述

题解

字符串最小循环节问题，这个在 $KMP$ 中的 $next[]$ 就可以获取。

首先，我们知道 $next[]$ 数组是控制失配时前跳的位置，所以模式串从第 $1$ 位到第 $next[n]$ 位与模式串第 $n - next[n]$ 位到第 $n$ 位是互相匹配的，如果这两个部分产生交集，则有可能是存在长度小于模式串长度的循环节，为啥是可能呢？因为此时我们还要判断是否整除，也就是 $n \ \%\ (n - next[n]) == 0$，如果整除，则存在，并且长度是 $n - next[n]$，否则，循环节也就是整个模式串。当然，这里我们并不需要判断是否产生交集，因为如果没有交集，后边是不可能整除的。

代码

#include <iostream>
#include <cstring>

using namespace std;

const int MAXN = 1e6;

char s[MAXN];
int nxt[MAXN];

void getnxt()
{
    nxt[0] = -1;
    int i = 0, j = -1;
    int len = (int)strlen(s);
    while (i < len)
    {
        if (s[i] == s[j] || j == -1)
        {
            i++;
            j++;
            nxt[i] = j;
        }
        else
        {
            j = nxt[j];
        }
    }
}

int main()
{
    while (~scanf("%s", s))
    {
        if (s[0] == '.')
        {
            break;
        }

        int len = (int)strlen(s);

        getnxt();

        if (len % (len - nxt[len]) == 0)
        {
            printf("%d\n", len / (len - nxt[len]));
        }
        else
        {
            printf("1\n");
        }
    }

    return 0;
}