Acwing 160. 匹配统计

阿轩在纸上写了两个字符串，分别记为A和B。

利用在数据结构与算法课上学到的知识，他很容易地求出了“字符串A从任意位置开始的后缀子串”与“字符串B”匹配的长度。

不过阿轩是一个勤学好问的同学，他向你提出了Q个问题：

在每个问题中，他给定你一个整数x，请你告诉他有多少个位置，满足“字符串A从该位置开始的后缀子串”与B匹配的长度恰好为x。

例如：A=aabcde，B=ab，则A有aabcde、abcde、bcde、cde、de、e这6个后缀子串，它们与B=ab的匹配长度分别是1、2、0、0、0、0。

因此A有4个位置与B的匹配长度恰好为0，有1个位置的匹配长度恰好为1，有1个位置的匹配长度恰好为2。

输入格式
第一行输入三个整数N,M,Q，分别表示A串长度、B串长度、问题个数。

第二行输入字符串A，第三行输入字符串B。

接下来Q行每行输入1个整数x，表示一个问题。

输出格式
输出共Q行，依次表示每个问题的答案。

数据范围
1≤N,M,Q,x≤200000
输入样例：
6 2 5
aabcde
ab
0
1
2
3
4
输出样例：
4
1
1
0
0

题解：哈希值+二分法；
（PS：有为什么输入字符串时不多预留一位会无输出？？？QAQ 如下错误代码）
80分代码

#include<iostream>
#include<cstring>
#include<string.h>
#include<cstdio>
using namespace std;
unsigned long long f[200005],fb[200005],p[200005];
int n,m,q,ans[200005];
char a[200005],b[200005];
int erfen(int be)
{
	if(a[be-1]!=b[0])return 0;
	int l=1,r=n-be+1,mid;
	while(l<r)
	{
		mid=(l+1+r)/2;
		if(f[be+mid-1]-f[be-1]*p[mid]==fb[mid])l=mid;
		else r=mid-1;
	}
	return r;
}
int main()
{
	cin>>n>>m>>q;
	scanf("%s",a);scanf("%s",b);p[0]=1;
	for(int i=1;i<=200000;i++)p[i]=p[i-1]*131;
	for(int i=1;i<=strlen(a);i++)f[i]=f[i-1]*131+a[i-1]-'a'+1;
	for(int i=1;i<=strlen(b);i++)fb[i]=fb[i-1]*131+b[i-1]-'a'+1;
	for(int i=1;i<=n;i++)
	{
		ans[erfen(i)]++;
	}
	int x;
	for(int i=1;i<=q;i++){scanf("%d",&x);printf("%d\n",ans[x]);}
	return 0;
}

---

AC代码

#include<iostream>
#include<cstring>
#include<string.h>
#include<cstdio>
using namespace std;
unsigned long long f[200005],fb[200005],p[200005];
int n,m,q,ans[200005];
char a[200005],b[200005];
int erfen(int be)
{
	if(a[be]!=b[1])return 0;
	int l=1,r=n-be+1,mid;
	while(l<r)
	{
		mid=(l+1+r)/2;
		if(f[be+mid-1]-f[be-1]*p[mid]==fb[mid])l=mid;
		else r=mid-1;
	}
	return r;
}
int main()
{
	cin>>n>>m>>q;
	scanf("%s",1+a);scanf("%s",1+b);
	p[0]=1;
	for(int i=1;i<=200000;i++)p[i]=p[i-1]*131;
	for(int i=1;i<=n;i++)f[i]=f[i-1]*131+a[i]-'a'+1;
	for(int i=1;i<=m;i++)fb[i]=fb[i-1]*131+b[i]-'a'+1;
	for(int i=1;i<=n;i++)
	{
		ans[erfen(i)]++;
	}
	int x;
	for(int i=1;i<=q;i++){scanf("%d",&x);printf("%d\n",ans[x]);}
	return 0;
}