撒斯泰克大学leb6d题

该题目来源于HDU - 3746

Step1 Problem:

给你一个字符串s, 问 s 变成周期串至少需要加几个字符。

Step2 Ideas:

个人习惯从0开始，next[0] = 0；
next[i] : 代表下标从 0 到 i 这个子串，后缀 = 前缀 最长长度（不包括自身）。
串的周期 T = len - next[len-1];
1.T == len, 添加 len 个字符，才能构成周期次数为 2 的周期串。
2.len % T == 0 && next[len-1] != 0, 添加 0 个字符，因为自身就是周期串
3.不满足上述两个条件，添加 T - len%T 个字符。

代码

#include "stdio.h";
#include "string.h";
int easyFunction(char* input);
int main() {
    int times=0;
    scanf("%d",&times);
    for (int i = 0; i < times; ++i) {
        char* str=new char[200001];
        scanf("%s",str);
        int A=easyFunction(str);
        int len=strlen(str);
        int T=len-A;
        if (T==len){
            printf("%d\n",len);
        }else if((len%T==0)&&(A!=0)){
            printf("0\n");
        } else{
            printf("%d\n",T-len%T);
        }
        delete [] str;
    }
}
int easyFunction(char* p){
    int m=strlen(p);
    int* next;
    next=new int[m];
    next[0]=0;
    int k=0,j=1;
    while(j<m){
        if (p[j]==p[k]){
            k=k+1;
            next[j]=k;
            j++;
        }else if(k==0){
            next[j]=0;
            j++;
        }else{
            k=next[k-1];
        }
    }
    return next[m-1];
}