阿轩在纸上写了两个字符串,分别记为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;
}