【哈希】The Preliminary Contest for ICPC Asia Shanghai 2019 - G - Substring

题目链接https://nanti.jisuanke.com/t/41415

---

题意

给出一个字符串s和Q个询问。每次询问一个字符串，问s有多少个子串和它匹配。匹配得定义是首尾字符一样且每个字符出现次数一样多。

---

题解

对于一次查询，可以$On$的滑窗扫一遍就行了。对与m个查询，可以把相同长度的放一起扫一遍，这样最坏复杂度是$On\sqrt{n}$。每次扫描用哈希和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;
}