首页 > 学院 > 开发设计 > 正文

【模板】 KMP算法

2019-11-14 10:47:02
字体:
来源:转载
供稿:网友

这里KMP有两种写法,一开始我是照着zyf2000的模板学的 跟她完全一样所以我直接先粘过来

#include<iostream>#include<cstring>#include<cstdio>using namespace std;const int max_W=10005;const int max_s=1000005;char W[max_W],s[max_s];int T[max_W];int t,len_W,len_s,ans;inline void calc_T(){ T[0]=-1; int j; for (int i=0;i<len_W;++i){ j=T[i]; while (j!=-1&&W[j]!=W[i]) j=T[j]; T[i+1]=++j; }}inline void calc_ans(){ int j=0; for (int i=0;i<len_s;++i){ while (j!=-1&&W[j]!=s[i]) j=T[j]; j++; if (j==len_W){ ans++; j=T[j]; } }}int main(){ scanf("%d/n",&t); while (t--){ gets(W); gets(s); len_W=strlen(W); len_s=strlen(s); memset(T,0,sizeof(T)); calc_T(); ans=0; calc_ans(); PRintf("%d/n",ans); }}

一开始我是这么写的,慢慢发现这样写实际对于有些题目并不适用,然后改成了另外一种写法,其实就是字符串从1开始计数,对于有些题比较方便。好像没有现成的模板也懒得再敲一遍了贴一下bzoj3620吧,事实上这道题的KMP体现的并不形象.noi2014的动物园比较形象,但是当时我还是照着第一种写的所以WA了好几次,改成第二种之后就不会出现我的错法了。

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>using namespace std;#define N 15005char s[N];int f[N];int k,l,ans,lim;int main(){ scanf("%s%d",s+1,&k); l=strlen(s+1);lim=l-k*2; for (int p=0;p<lim;p++)//枚举左端点,对每一个左端点做KMP { for (int j=0,i=2;i+p<l;i++)//处理next(f)数组 { while (j&&s[j+p+1]!=s[i+p])j=f[j]; if (s[i+p]==s[j+p+1])j++;f[i]=j; } for (int j=0,i=k+1;i+p<=l;i++)//类似noi2014的动物园 { while (j&&s[i+p]!=s[j+p+1])j=f[j]; if (s[i+p]==s[j+p+1])j++; while ((j<<1)>=i)j=f[j];if (j>=k)ans++;//当前缀与后缀都>=k即j>=k时并且<=i>>1时计入答案 } } cout<<ans; return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表