拓展kmp的应用

Smiling & Weeping

　　　　　　　　　　　　　　　　---- 我与月亮，进行了一次深夜谈话

　　　　　　　　　　　　　　　　　　它与我谈论太阳，而我与它谈论你。

 

题目链接：P3435 [POI2006] OKR-Periods of Words - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

思路：其实也就是kmp拓展，求s与s的每个后缀的LCP（最长公共前缀），但是这样超时，需要一个小小的优化（记录跳转即step[k]，跳到可以上一个可以实现的地方，尽量保持复杂度在线性），同时s[i] == s[0]时，直接使ans += i;

Talk is cheap , show me the code

#include<bits/stdc++.h>
using namespace std;
int n , Next[1000100] , step[1000100];
char str[1000100];
void getNext(char *c){
    int len = strlen(c);
    int p = 0 , k = 1 , l;
    Next[0] = len;
    while(p+1 < len && c[p] == c[p+1]) p++;
    Next[1] = p;
    for(int i = 2; i < len; i++){
        p = k+Next[k]-1;
        l = Next[i-k];
        if(i+l <= p) Next[i] = l;
        else{
            int j = max(0 , p-i+1);
            while(i+j < len && c[i+j] == c[j]) j++;
            Next[i] = j;
            k = i;
        }
    } 
}
int main()
{
    long long ans = 0;
    int q=0 , k;                      // q代表周期，k代表最远到达的距离
    scanf("%d",&n); scanf("%s",str);
    getNext(str);
    for(int i = 1; i < n; i++){
        if(str[i] == str[0]){
            ans += i;
        } else {
            int r = i-1 , tem = r;
            while(Next[r] < i-r+1 && r >= (i+1)/2){
                if(step[r] && step[r] < r) // 警惕bug，可能有step[r]==r,陷入死循环
                    r = step[r];
                else r--;
            }
            if(r >= (i+1)/2 && Next[r] >= i-r+1) ans += r;
            step[tem] = r;
        }
    }
    printf("%lld\n",ans);
    return 0;
}

一棵树，一块岩石，一朵云

文章到此结束，我们下次再见