HDU 1358 Period

Posted 2020-06-13 shiyicode

题目链接：[kuangbin带你飞]专题十六 KMP & 扩展KMP & Manacher E - Period

题意

给一字符串，求其所有完整循环的前缀与循环节的长度。
例：aaa
长度2前缀，循环节为a，个数为2
长度3前缀，循环节为a，个数为3

思路

kmp求出字符串前后缀重复数，遍历所有前缀子串进行下面操作：
字符串前后缀重复数next[L]，则循环节的长度为L-L%next[L]，如果L%循环节长度为0，则说明是完整循环，输出解。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <vector>

using namespace std;

char a[1000009];
int p[1000009];

void init(int m)
{
    memset(p, 0, sizeof(p));
    for(int i=1, k=0; i<m; i++)
    {
        while(k>0 && a[i]!=a[k]) k = p[k-1];
        if(a[i] == a[k]) k++;
            p[i] = k;
    }
}

void solve(int n)
{
    init(n);
    for(int i=1; i<n; i++)
    {
        int l = i+1-p[i];
        if(l != i+1 && (i+1)%l == 0)
            printf("%d %d\n", i+1, (i+1)/l);
    }
}

int main()
{
    int n, T = 1;
    while(~scanf("%d", &n) && n)
    {
        scanf("%s", a);
        printf("Test case #%d\n", T++);
        solve(n);
        printf("\n");
    }
    return 0;
}