CODEVS——T 1404 字符串匹配

http://codevs.cn/problem/1404/

时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master

题解

 查看运行结果

 

 

题目描述 Description

给你两个串A，B，可以得到从A的任意位开始的子串和B匹配的长度。
给定K个询问，对于每个询问给定一个x，求出匹配长度恰为x的位置有多少个。
N，M，K<=200000

输入描述 Input Description

第一行三个数 N，M，K，表示A的长度、B的长度和询问数。
第二行为串A。
第三行为串B。
接下来K行，每行1个数X。

输出描述 Output Description

对于每个询问输出一个数。

样例输入 Sample Input

6 2 2
aabcde
ab
0
2

样例输出 Sample Output

4
1

数据范围及提示 Data Size & Hint

各个测试点1s

kmp得出ans[],然后考虑将串分解，，倒序累加ans[next[i]],,此时的ans[i]是>=i位的数量，所以最后还要处理为ans[i]-=ans[i+1]

 #include <algorithm>
 #include <cstring>
 #include <cstdio>

 using namespace std;

 const int N(+);
 int n,m,k,lb,la;
 int p[N],ans[N];
 char a[N],b[N];

 inline void Get_next()
 {
     for(int i=,j=;i<=lb;p[i++]=j)
     {
         for(;b[i]!=b[j+]&&j>;) j=p[j];
         if(b[i]==b[j+]) j++;
     }
 }
 inline void kmp()
 {
     for(int i=,j=;i<=la;i++)
     {
         for(;a[i]!=b[j+]&&j>;) j=p[j];
         if(a[i]==b[j+]) j++;
         ans[j]++;
     }
 }

 int main()
 {
     scanf("%d%d%d%s%s",&n,&m,&k,a+,b+);
     la=strlen(a+); lb=strlen(b+);
     Get_next(); kmp();
     for(int i=lb;i>;i--) ans[p[i]]+=ans[i];
     for(int i=;i<lb;i++) ans[i]-=ans[i+];
     for(int pos;k--;)
     {
         scanf("%d",&pos);
         printf("%d\n",ans[pos]);
     }
     return ;
 }