UVALive - 3026 Period

题意:给定一个长度为n的字符串S,求它每个前缀的最短的循环节,换句话来说,对于每个i,求一个最大的整数k>1,使得S的前i个字符组成的前缀是某个字符串复制k次得到的,

输出所有存在的k的i和对应的k

思路:如果这i个字符组成一个周期串,那么循环节就是(i-f[i]),f[i]对应KMP的next[]数组

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 1000005;

char str[MAXN];
int f[MAXN];

int main(){
    int n,t = 0;
    while (scanf("%d",&n) != EOF && n){
        scanf("%s",str);
        f[0] = f[1] = 0;
        for (int i = 1; i < n; i++){
            int j = f[i];
            while (j && str[i] !=str[j])
                j = f[j];
            f[i+1] = (str[i] == str[j]) ? j+1 : 0;
        }
        printf("Test case #%d\n",++t);
        for (int i = 2; i <= n; i++)
            if (f[i] > 0 && i % (i-f[i]) == 0)
                printf("%d %d\n",i,i/(i-f[i]));
        printf("\n");
    }
    return 0;
}

你可能感兴趣的:(UVALive - 3026 Period)