[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;
}

猜你喜欢

转载自www.cnblogs.com/MYsBlogs/p/11826401.html