Codeforces Round #291 (Div. 2) C. Watto and Mechanism（哈希）

题目链接

题意：
给定n个串，然后给出m个询问。
对于每个询问，判断 能否 更改一个字符使得其变为n个串中的一个。
串只有 a b c 三种字符。
hash乱搞+枚举。

题解：
好像没写过hash的博客，就拿这个来写一下。
⑧过单hash被卡掉了，双hash乱搞一下就好啦。
为了不显得那么乱，把两个hash分开了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<set>
#define ll long long
#define llu unsigned ll
#define pr make_pair
#define pb push_back
#define ui unsigned int
#define lc (cnt<<1)
#define rc (cnt<<1|1)
#define len(x)  (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
#define forhead(x) for(int i=head[(x)];i;i=nt[i])
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)>(y)?(y):(x))
using namespace std;
const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1000000007;
const double eps=1e-8;
const double pi=acos(-1.0);
const int maxn=600100;
const int maxm=100100;
const int up=100000;
const int hashp=13331;
const int hashpp=131;
//给定n个串，然后给出m个询问。
//对于每个询问，判断 能否 更改一个字符使得其变为n个串中的一个
//串只有abc三种字符。
//hash乱搞+枚举

llu p[maxn],hs[maxn];
llu pp[maxn],hhs[maxn];
char s1[maxn],s2[maxn];
int n,m;
set<llu>se,see;

llu get(void)
{
    int len=strlen(s1);
    llu ans=0;
    for(int i=0;i<len;i++)
        ans=ans*hashp+s1[i];
    return ans;
}

llu get2(void)
{
    int len=strlen(s1);
    llu ans=0;
    for(int i=0;i<len;i++)
        ans=(ans*hashpp+s1[i])%mod;
    return ans;
}

void init(void)
{
    p[0]=1;
    for(int i=1;i<maxn;i++)
        p[i]=p[i-1]*hashp;
    pp[0]=1;
    for(int i=1;i<maxn;i++)
        pp[i]=pp[i-1]*hashpp%mod;
}

bool check(void)
{
    int len=strlen(s2+1);
    for(int i=1;i<=len;i++)
        hs[i]=hs[i-1]*hashp+s2[i];
    llu pre,last;
    for(int i=1;i<=len;i++)
    {
        pre=hs[i-1]*p[len-i+1];
        last=hs[len]-hs[i]*p[len-i];
        if(s2[i]!='a'&&se.find(pre+'a'*p[len-i]+last)!=se.end()) return true;
        if(s2[i]!='b'&&se.find(pre+'b'*p[len-i]+last)!=se.end()) return true;
        if(s2[i]!='c'&&se.find(pre+'c'*p[len-i]+last)!=se.end()) return true;
    }
    return false;
}

bool check2(void)
{
    int len=strlen(s2+1);
    for(int i=1;i<=len;i++)
        hhs[i]=(hhs[i-1]*hashpp+s2[i])%mod;
    llu pre,last;
    for(int i=1;i<=len;i++)
    {
        pre=(hhs[i-1]*pp[len-i+1])%mod;
        last=(hhs[len]-hhs[i]*pp[len-i]%mod+mod)%mod;
        if(s2[i]!='a'&&see.find((pre+'a'*pp[len-i]+last)%mod)!=see.end()) return true;
        if(s2[i]!='b'&&see.find((pre+'b'*pp[len-i]+last)%mod)!=see.end()) return true;
        if(s2[i]!='c'&&see.find((pre+'c'*pp[len-i]+last)%mod)!=see.end()) return true;
    }
    return false;
}

int main(void)
{
    init();
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s1);
        se.insert(get());
        see.insert(get2());
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%s",s2+1);
        if(check()&&check2()) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}