题目链接https://nanti.jisuanke.com/t/41415
题意
给出一个字符串s和Q个询问。每次询问一个字符串,问s有多少个子串和它匹配。匹配得定义是首尾字符一样且每个字符出现次数一样多。
题解
对于一次查询,可以 O n On On的滑窗扫一遍就行了。对与m个查询,可以把相同长度的放一起扫一遍,这样最坏复杂度是 O n n On\sqrt{n} Onn。每次扫描用哈希和map维护答案。map要用count,不然会炸空间。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+7;
const ll mod=1e9+7;
unordered_map<ll,int>mp[26][26];
unordered_map<ll,bool>ask[26][26];
struct Que{
int id,len;
int bg,ed;
ll hs;
bool operator<(const Que k)const{return len<k.len;}
}q[N];
int t,n,sz;
char s[N],ss[N];
ll base[30],bb,be;
int ans[N];
void solve(int len){
if(len>sz) return;
int i=0,j=-1;
ll now=0;
while(j+1<len) j++,now=(now+base[s[j]-'a'])%mod;
if(ask[s[i]-'a'][s[j]-'a'].count(now)) mp[s[i]-'a'][s[j]-'a'][now]++;
while(j+1<sz){
now=(now-base[s[i]-'a']+mod)%mod; i++;
j++; now=(now+base[s[j]-'a'])%mod;
if(ask[s[i]-'a'][s[j]-'a'].count(now)) mp[s[i]-'a'][s[j]-'a'][now]++;
}
}
int main()
{
scanf("%d",&t);
base[0]=N;
for(int i=1;i<26;i++) base[i]=base[i-1]*N%mod;
while(t--){
for(int i=0;i<26;i++) for(int j=0;j<26;j++) mp[i][j].clear(),ask[i][j].clear();
scanf("%s",s);
sz=strlen(s);
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%ss",ss);
q[i].len=strlen(ss);
q[i].id=i;
q[i].bg=ss[0]-'a';
q[i].ed=ss[q[i].len-1]-'a';
q[i].hs=0;
for(int j=0;j<q[i].len;j++) q[i].hs=(q[i].hs+base[ss[j]-'a'])%mod;
ask[q[i].bg][q[i].ed][q[i].hs]=true;
}
sort(q+1,q+1+n);
for(int i=1;i<=n;i++){
if(q[i].len!=q[i-1].len) solve(q[i].len);
ans[q[i].id]=mp[q[i].bg][q[i].ed][q[i].hs];
}
for(int i=1;i<=n;i++){
printf("%d\n",ans[i]);
}
}
return 0;
}