【codevs1404】字符串匹配 KMP
题目描述
给你两个串A,B,可以得到从A的任意位开始的子串和B匹配的长度。
给定K个询问,对于每个询问给定一个x,求出匹配长度恰为x的位置有多少个。
N,M,K<=200000
输入
第一行三个数 N,M,K,表示A的长度、B的长度和询问数。
第二行为串A。
第三行为串B。
接下来K行,每行1个数X。
输出
对于每个询问输出一个数。
样例输入
6 2 2
aabcde
ab
0
2
样例输出
4
1
题解
KMP先求出next数组,匹配一遍,统计一下某长度出现的次数num。
这是以每个字符结尾的出现次数,然而落下了一部分。
为了得到总数,需要从后向前执行num[next[i]]+=num[i]可以得到能够匹配某长度的个数。
而题目要求是“恰好匹配”,所以答案为num[i]-num[i+1]。
KMP细节太多了。。。
#include <cstdio> #include <cstring> char A[200001] , B[200001]; int next[200001] , num[200001]; int main() { int n , m , k , i , j; scanf("%d%d%d%s%s" , &n , &m , &k , A , B); for(i = 1 ; i < m ; i ++ ) { j = next[i]; while(j && B[i] != B[j]) j = next[j]; if(B[i] == B[j]) next[i + 1] = j + 1; else next[i + 1] = 0; } for(i = j = 0 ; i < n ; i ++ ) { while(j && A[i] != B[j]) j = next[j]; if(A[i] == B[j]) j ++ ; num[j] ++ ; } for(i = m ; i > 0 ; i -- ) num[next[i]] += num[i]; while(k -- ) { scanf("%d" , &i); printf("%d\n" , num[i] - num[i + 1]); } return 0; }