hdu 1358 Period （KMP求循环次数）

Problem - 1358

　　KMP求循环节次数。题意是，给出一个长度为n的字符串，要求求出循环节数大于1的所有前缀。可以直接用KMP的方法判断是否有完整的k个循环节，同时计算出当前前缀的循环节的个数。

代码如下：

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

using namespace std;

const int N = 1111111;
char buf[N];
int next[N];

void getNext(char *str, int len) {
    int i = 0, j = -1;
    next[0] = -1;
    while (i <= len) {
        while (j > -1 && str[i] != str[j]) j = next[j];
        i++, j++;
        next[i] = j;
    }
}

typedef pair<int, int> PII;
PII rec[N];

int main() {
    int n, cas = 1;
    while (cin >> n && n) {
        cin >> buf;
        getNext(buf, n);
//        for (int i = 0; i <= n; i++) cout << next[i] << ' '; cout << endl;
        int cnt = 0;
        for (int i = 1; i <= n; i++) {
            if (i % (i - next[i])) continue;
            if (i / (i - next[i]) == 1) continue;
            rec[cnt++] = PII(i, i / (i - next[i]));
        }
        sort(rec, rec + cnt);
        printf("Test case #%d
", cas++);
        for (int i = 0; i < cnt; i++) {
            printf("%d %d
", rec[i].first, rec[i].second);
        }
        puts("");
    }
    return 0;
}

——written by Lyon