题目大意:
给定(n)个字符串和(q)个询问
每次询问在这(n)个字符串中,有多少个字符串同时满足
- 字符串a是它的前缀
- 字符串b是它的后缀
(n,qle 50000,|S|le 100)
解法:
AC的人中唯一一个hash的 其实是不会trie树
考虑 长度相等的字符串a的查询(设长度为len) 放一起处理
把字符串b的长度塞进容器中(要去重)
那么只用 将给定的(n)个字符串长度为len的前缀与容器中每一个长度的后缀塞入map中+1
最后查询一下map就可以了
理论复杂度(O(n|S|^2log_2n)),但是实际远远到不了这么大
毕竟出题人肯定想不到这种解法
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define dwn(i,a,b) for(int i=(a);i>=(b);--i)
using namespace std;
typedef long long ll;
const int mod=1999993;
int n,Q,ans[50010],p[110]={1,13};
int hash[50010][110],len[50010];
char s[50010][110];
map<int,int> mp[2000010];
int ind=0,st[50010];
struct node
{
char s1[110],s2[110];
int len1,len2,hash1,hash2,id;
}q[50010];
bool cmp(node a,node b)
{
return a.len1==b.len1?a.len2<b.len2:a.len1<b.len1;
}
inline int read()
{
int x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-') f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
x=(x<<3)+(x<<1)+(ch^48);
ch=getchar();
}
return x*f;
}
inline int get(int i,int l,int r)
{
return ((hash[i][r]-(ll)hash[i][l-1]*p[r-l+1])%mod+mod)%mod;
}
int main()
{
n=read(),Q=read();
rep(i,2,102) p[i]=(ll)p[i-1]*p[1]%mod;
rep(i,1,n)
{
scanf("%s",s[i]+1),len[i]=strlen(s[i]+1);
for(int j=1;j<=len[i];++j)
{
hash[i][j]=((ll)hash[i][j-1]*p[1]+s[i][j]-'a'+1)%mod;
}
}
rep(i,1,Q)
{
q[i].id=i;
scanf("%s%s",q[i].s1+1,q[i].s2+1);
q[i].len1=strlen(q[i].s1+1),q[i].len2=strlen(q[i].s2+1);
rep(j,1,q[i].len1) q[i].hash1=((ll)q[i].hash1*p[1]+q[i].s1[j]-'a'+1)%mod;
rep(j,1,q[i].len2) q[i].hash2=((ll)q[i].hash2*p[1]+q[i].s2[j]-'a'+1)%mod;
}
sort(q+1,q+Q+1,cmp);
for(int i=1,j;i<=Q;i=j+1)
{
j=i;
ind=0;
mp[q[i].hash1].clear();
st[++ind]=q[i].len2;
while(j+1<=Q&&q[i].len1==q[j+1].len1)
{
++j;
mp[q[j].hash1].clear();
st[++ind]=q[j].len2;
}
ind=unique(st+1,st+ind+1)-st-1;
rep(k,1,n)
{
if(len[k]>=q[i].len1)
{
for(int t=1;t<=ind&&len[k]>=st[t];++t)
{
++mp[get(k,1,q[i].len1)][get(k,len[k]-st[t]+1,len[k])];
}
}
}
rep(k,i,j)
{
ans[q[k].id]+=mp[q[k].hash1][q[k].hash2];
}
}
rep(i,1,Q)
{
printf("%d
",ans[i]);
}
return 0;
}