一辈子要学N次的kmp

CF 471D MUH and Cube Walls

• 题意：找相同走势的数列的个数
• 思路：做差，然后kmp

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<=b;i++)
const int MAXN=2e5+10;
int a[MAXN],b[MAXN];
int nxt[MAXN];
int kmp(int n,int w){
    int res=0;
    nxt[1]=0;
    for(int i=2,j=0;i<=w;i++){
        while(j!=0&&b[i]!=b[j+1])j=nxt[j];
        if(b[i]==b[j+1])j++;
        nxt[i]=j;
    }
    for(int i=1,j=0;i<=n;i++){
        while(j!=0&&(j==w||a[i]!=b[j+1]))j=nxt[j];
        if(a[i]==b[j+1])j++;
        if(j==w)res++;
    }
    return res;
}
int main(){
    int n,w;
    scanf("%d%d",&n,&w);
    rep(i,1,n){
        scanf("%d",&a[i]);
        a[i-1]=a[i-1]-a[i];
    }
    rep(i,1,w){
        scanf("%d",&b[i]);
        b[i-1]=b[i-1]-b[i];
    }
    if(w==1){
        printf("%d",n);
        return 0;
    }
    n--;
    w--;
    printf("%d",kmp(n,w));
    return 0;
}

POJ 2406 Power Strings

• 题意：求循环节
• 思路：kmp next数组 len/（len-next[len]）//on 1 base && 整除

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define rep(i,a,b) for(int i=a;i<=b;i++)
const int MAXN=1e6+10;
char s[MAXN];
int nxt[MAXN];
int main(){
    while(scanf("%s",s),s[0]!='.'){
        int len=strlen(s);
        if(len==0){
            puts("0");
            continue;
        }
        nxt[0]=-1;
        for(int i=1,j=-1;i<len;i++){
            while(j!=-1&&s[i]!=s[j+1])j=nxt[j];
            if(s[i]==s[j+1])j++;
            nxt[i]=j;
        }
        if(len%(len-1-nxt[len-1])==0){
            printf("%d\n",len/(len-1-nxt[len-1]));
        }else{
            puts("1");
        }
    }
    return 0;
}