题意
给你两个串A,B,可以得到从A的任意位开始的子串和B匹配的长度。
给定K个询问,对于每个询问给定一个x,求出匹配长度恰为x的位置有多少个。
N,M,K<=200000
题解
KMP是正解,这里就用Hash+二分水一下,复杂度
。
具体做法如下:
首先将A、B两个串全部用同一种Hash方法进行Hash,然后需要统计匹配
个字符有多少种情况。 不难想到匹配满足单调性:枚举A串的起点,二分匹配的长度mid,若
与
匹配,向右二分匹配的位置,否则向左二分匹配的位置。最后统计结果即可。
代码
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
const int nmax = 2e5 + 100;
const int INF = 0x3f3f3f3f;
const ull p = 67;
int lenstr,lentar;
int ans[nmax],q,x;
char str[nmax],tar[nmax];
ull hashstr[nmax],hashtar[nmax],pp[nmax] = {0,p};
inline ull hashsub(int l, int r){
return hashstr[r] - hashstr[l-1] * pp[r-l+1];
}
int main() {
for(int i = 2;i<nmax;++i) pp[i] = pp[i-1] * p;
scanf("%d %d %d",&lenstr,&lentar,&q);
scanf("%s %s",str+1,tar+1); hashstr[1] = str[1], hashtar[1] = tar[1];
for(int i = 2;i<=lenstr;++i) hashstr[i] = hashstr[i-1] * p + str[i];
for(int i = 2;i<=lentar;++i) hashtar[i] = hashtar[i-1] * p + tar[i];
for(int i = 1;i<=lenstr;++i){
int l = 1, r = min(lentar,lenstr - i + 1), mid,pos = 0;
while(l <= r){
mid = (l+r)>>1;
if(hashsub(i,i+mid-1)== hashtar[mid]) pos = mid, l = mid + 1;
else r = mid-1;
}
ans[pos] ++;
}
while(q--){scanf("%d",&x);printf("%d\n",ans[x]);}
return 0;
}