【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;
}