洛谷P3375 [模板]KMP字符串匹配

To 洛谷.3375 KMP字符串匹配

题目描述

如题，给出两个字符串s1和s2，其中s2为s1的子串，求出s2在s1中所有出现的位置。

为了减少骗分的情况，接下来还要输出子串的前缀数组next。如果你不知道这是什么意思也不要问，去百度搜[kmp算法]学习一下就知道了。

输入输出格式

输入格式：

第一行为一个字符串，即为s1（仅包含大写字母）

第二行为一个字符串，即为s2（仅包含大写字母）

输出格式：

若干行，每行包含一个整数，表示s2在s1中出现的位置

接下来1行，包括length(s2)个整数，表示前缀数组next[i]的值。

输入输出样例

输入样例#1：

ABABABC
ABA

输出样例#1：

1
3
0 0 1

说明

时空限制：1000ms,128M

数据规模：

设s1长度为N，s2长度为M

对于30%的数据：N<=15，M<=5

对于70%的数据：N<=10000，M<=100

对于100%的数据：N<=1000000，M<=1000

样例说明：

所以两个匹配位置为1和3，输出1、3

代码：

 #include<cstdio>
 #include<cstring>
 using namespace std;
 char s1[],s2[];
 int len1,len2,fai[];
 void getfail()
 {
     fai[]=fai[]=;
     for(int i=;i<len2;i++)
     {
         int j=fai[i];
         while(j&&s2[i]!=s2[j]) j=fai[j];
         fai[i+]= s2[i]==s2[j] ? j+ : ;
     }
 }
 void kmp()
 {
     int j=;
     for(int i=;i<len1;i++)
     {
         while(j&&s1[i]!=s2[j]) j=fai[j];
         if(s1[i]==s2[j]) ++j;
         if(j==len2)
           printf("%d\n",i-len2+);
     }
 }
 int main()
 {
     scanf("%s",s1);
     scanf("%s",s2);
     len1=strlen(s1);
     len2=strlen(s2);
     getfail();
     kmp();
     for(int i=;i<=len2;i++)
       printf("%d ",fai[i]);
     return ;
 }

[2018.4.4] ：更新后的模板

 #include <cstdio>
 #include <cstring>
 const int N=1e6+;

 int len,fail[N];
 char p[N],s[N];

 void Get_fail()
 {
     fail[]=;
     for(int i=,j; i<len; ++i)
     {
         j=fail[i];
         while(s[i]!=s[j]&&j) j=fail[j];
         fail[i+]=s[i]==s[j]?j+:;
     }
 }
 void KMP()
 {
     for(int i=,j=,l=strlen(p); i<l; ++i)
     {
         while(p[i]!=s[j]&&j) j=fail[j];
         if(p[i]==s[j]) ++j;
         if(j==len) printf("%d\n",i-j+);
     }
     for(int i=; i<=len; ++i) printf("%d ",fail[i]);
 }

 int main()
 {
     scanf("%s%s",p,s), len=strlen(s), Get_fail(), KMP();
     return ;
 }