HDU 3746 Cyclic Nacklace (KMP找循环节)

题目链接：HDU 3746

Sample Input

3
aaa
abca
abcde

Sample Output

0
2
5

Author

possessor WC

Source

HDU 3rd “Vegetable-Birds Cup” Programming Open Contest

Solution

题意

给定一个字符串，问至少需要在末尾添加多少个字符使得字符串循环。

思路

KMP

设前缀函数为 \(\pi()\)，字符串长度为 \(n\)，下标从 \(1\) 开始。

最小循环节为 \(k = n - \pi(n)\)。

如果 \(n \% k=0\)，那么字符串已经循环。

否则还需要添加 \(k - n \% k\) 个字符。

Code

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int n, pi[maxn];
char str[maxn];

void prefix_function() {
    int j = 0;
    for (int i = 2; i <= n; ++i) {
        while (j > 0 && str[j+1] != str[i]) j = pi[j];
        if (str[j+1] == str[i]) j++;
        pi[i] = j;
    }
}

int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        scanf("%s", str + 1);
        n = strlen(str + 1);
        prefix_function();
        if(pi[n] == 0) {
            printf("%d\n", n);
        } else {
            int k = n - pi[n];
            if(n % k == 0) {
                printf("0\n");
            } else {
                printf("%d\n", k - n % k);
            }
        }
    }
    return 0;
}