字符串哈希(双模数哈希)-Games on a CD-CodeForces-727E

字符串哈希(双模数哈希)-Games on a CD-CodeForces-727E

题意：

$$\begin{gathered} 给定一个由n个长度为k的子串组成的n\times k的字符串s，以及g个长度为k的字符串。 \\ 判断：是否能够在这g个串中，选择n个串拼接成s，若能，输出依次选择的顺序。 \end{gathered}$$

数据范围：

$n,k\in [1,100000]，且n\times k<=1000000；g\in [n,100000]，且g\times k<=2000000。$

Sample Input:
3 1
abc
4
b
a
c
d

Sample Output:
YES
2 1 3 


Sample Input:
4 2
aabbccdd
4
dd
ab
bc
cd

Sample Output:
NO

题解:

$$\begin{gathered} 首先将s先复制，拓展成“环”。 \\ 接着对输入的g个字符串分别哈希，将哈希值存入到map当中，再分枚举s的起点，求s的n个长度为k的子串的哈希值。 \\ 若这n个哈希值均在map中，说明可以构成，按顺序输出即可。 \end{gathered}$$

$$\begin{gathered} 需要说明的是： \\ ①、由于k的长度可能会比较小，冲突的概率会比较大，所以取双模数，用pair存储哈希值。 \\ ②、枚举s的起点时，仅需枚举k次，这是因为每个子串的长度均为k，从k+1处枚举与从1位置枚举等价。 \\ ③、两种原因会导致失败，一是s中某个长度为k的子串的哈希值不在这g个字符串中； \\ 二是g个字符串中的每个串仅能使用一次， \\ 若同一起点开始，枚举到的子串的哈希值在map中且已经被使用过，也是不满足条件的。 \\ 这个问题可以用数组st来记录同一起点枚举的过程中，各个被使用的子串对应的这个起点。 \\ ④、bool函数忘记return会返回一个不确定的值。 \end{gathered}$$

---

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#define ll long long
#define P pair<ll,ll>
#define x first
#define y second
using namespace std;
const int N=2e6+10;
const int base=131;
const int mod1=1e9+7;
const int mod2=1e9+9;
int n,k,g;
ll h[N][3],p[3];
char s[N],t[N];
map<P,int> M;
int ans[N],st[N];

bool check(int pos)
{
    int l=pos,r=pos+k-1;
    for(int i=1;i<=n;i++)   
    {
        ll h1=(h[r][1]-h[l-1][1]*p[1]%mod1+mod1)%mod1;
        ll h2=(h[r][2]-h[l-1][2]*p[2]%mod2+mod2)%mod2;
        P tmp=make_pair(h1,h2);
        if(!M.count(tmp)||st[M[tmp]]==pos) return false;
        ans[i]=M[tmp];
        st[ans[i]]=pos;
        l+=k,r+=k;
    }
    return true;
}

int main()
{
    scanf("%d%d",&n,&k);
    scanf("%s",s+1);
    int len=strlen(s+1);
    for(int i=len+1;i<=2*len;i++) s[i]=s[i-len];   ///
    len=2*len,s[len+1]='\0';

    p[1]=p[2]=1;
    for(int i=1;i<=k;i++)
        p[1]=p[1]*base%mod1,
        p[2]=p[2]*base%mod2;

    for(int i=1;i<=len;i++)
    {
        h[i][1]=(h[i-1][1]*base%mod1+s[i]-'a')%mod1,
        h[i][2]=(h[i-1][2]*base%mod2+s[i]-'a')%mod2;
    }

    scanf("%d",&g);
    for(int i=1;i<=g;i++)  ///计算g个子串的哈希值
    {
        scanf("%s",t+1);
        ll h1=0,h2=0;
        for(int j=1;j<=k;j++)
            h1=(h1*base%mod1+t[j]-'a')%mod1,
            h2=(h2*base%mod2+t[j]-'a')%mod2;

        M[make_pair(h1,h2)]=i;
    }

    for(int i=1;i<=k;i++)  ///枚举起点位置
        if(check(i))
        {
            puts("YES");
            for(int i=1;i<=n;i++) printf("%d ",ans[i]);
            puts("");
            return 0;
        }

    puts("NO");

    return 0;
}