- 题意:找相同走势的数列的个数
- 思路:做差,然后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;
}
- 题意:求循环节
- 思路: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;
}