Cyclic Nacklace HDU - 3746 (前缀数组 kmp)

Cyclic Nacklace HDU - 3746
题意:给你一串字符串。可以在这个字符串左右加上任意字符，使最后求得的字符串可以构成循环。

解题思路：

用前缀数组可以求得初始字符串的最小循环节。
最小循环节=（原字符串长度-末尾前缀数组值）
然后进行讨论：

• 如果最小循环节=原子符串长度.说明末尾前缀数组值为0.说明原串中，只能通过复制一次原串得到目标字符串。 ans=原串长度
• 如果原串长度%最小循环节=0.说明可以通过循环原串中某前缀就可以构成原串，原串已经完成循环。 ans=0
• 如果不是上面两种情况。说明原串中，有部分不在循环中。可以求得这部分长度为最小循环节-原串长度%最小循环节。我们加上这部分就可以构成循环，满足条件。** ans=最小循环节-原串长度%最小循环节

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
//#pragma GCC optimize(3,"Ofast","inline")
#define LL long long
#define MT(a,b) memset(a,b,sizeof(a))
const int mod=1000007;
const int maxn=1e6+5;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
char s[maxn];
int pi[maxn];
int get_pi(char *str){
    int len=strlen(str);
    pi[0]=0;
    for (int i=1;i<len;i++){
        int j=pi[i-1];
        while (j>0&&str[j]!=str[i]) j=pi[j-1];
        if (str[j]==str[i]) j++;
        pi[i]=j;
    }
    int tmp=len-pi[len-1];
    if (!pi[len-1]) return len;
    else if (!(len%tmp)) return 0;
    else return tmp-len%tmp;
}
int main (){
    int T;
    scanf("%d",&T);
    while (T--){
        scanf("%s",s);
        printf("%d\n",get_pi(s));
    }
    return 0;
}