【codevs】 1404 字符串匹配(KMP)

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

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N = 200010;int num[N];void get_next(char *t,int *next){   int i=0;//Prefix 前缀   int j=1;//Postfix 后缀   next[0] = 0;//自定义的，0和1都从0开始匹配   next[1] = 0;   while(t[j] != '\0')   {       if(t[i] == t[j])//若前后字符匹配，则向前推进       {           i++;           j++;           next[j] = i;       }       else       {           i = next[i];///前后字符不匹配，则回溯。注意，此时是i和j不匹配，因此，根据next数组定义，要回溯到next[i]的值       }       if(0 == i)//当回溯到首字符时，单独进行处理       {           if(t[i] == t[j])           {               next[++j] = ++i;           }           else next[++j] = i;       }   }}void KMP(int *next,char *ch1,char *ch,int m,int n){    get_next(ch1,next);int j = 0;for(int i = 0; i < n; ++i){while(j && ch[i] != ch1[j]) j = next[j];if(ch1[j] == ch[i]) j++;num[j]++;/*KMP中计算的num[]是统计以i为尾的匹配长度的数量，会将匹配失败的位置统计到当前匹配长度中，而不是统计到真的更小的匹配中，但总数依旧是不变的，且多统计的长度始终>=少统计的长度，于是我们倒序进行这个操作：num[next[j]]+=num[j]（next[]为失配指针）因为匹配的时候，总是让j=next[j]，所以所有匹配到的>next[j]的位置，都会加入num[next[j]]的统计，此时所有可能的匹配长度>=L的位置都已经统计入num[L]中然后那些漏掉的就补回来了，多算的没得补（也就是刚才的统计错误被修正了）于是长度为x的答案就是num[x]-num[x+1]*/}}int main(){char ch[N], ch1[N];int n, m, k;int next[N];cin>>n>>m>>k;cin>>ch>>ch1;int x;KMP(next,ch1,ch,m,n);    for(int i = m; i > 0; i--)        num[next[i]] += num[i];    for(int i = 0; i <= m; i++)        num[i] -= num[i + 1];for(int i = 1; i <= k; ++i){cin>>x;cout << num[x] << endl;}return 0;}