[JZOI3887]字符串查询hash解法

题目大意：

给定\(n\)个字符串和\(q\)个询问
每次询问在这\(n\)个字符串中，有多少个字符串同时满足

1. 字符串a是它的前缀
2. 字符串b是它的后缀

\(n,q\le 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\n",ans[i]);
	}
	return 0;
}