【JZOJ3870】【NOIP2014八校联考第4场第1试10.19】单词检索(search)

Description

小可可是学校图书馆的管理员，现在他接手了一个十分棘手的任务。

由于学校需要一些材料，校长需要在文章中检索一些信息。校长一共给了小可可N篇文章，每篇文章为一个字符串。现在，校长需要他找到这样的单词，它至少在这N篇文章中的M篇文章里出现过，且单词长度为L。可是，工作量十分庞大，但校长又急需小可可完成这项任务。

现在他向你求助，需要你编写程序完成这项艰巨的任务。

Data Constraint

对于20%的数据有1≤N,M≤10；

对于60%的数据有1≤N,M≤100；

对于100%的数据有1≤N,M≤2000，L≤1000。每篇文章长度不大于1000，均有小写字母组成。

Solution

这道题一看就想到用哈希表来判重。我们对于每个输入字符串s，我们记录a[i]表示s的前i位的哈希值。那么当你选择了i-l+1~i的子串时，它的字符串哈希值就为a[i]-a[i-l]*s[l]。先在一个哈希数组中判断一下，假设某个子串在哈希数组中出现了超过1次，就把它加入到总的哈希数组中（为了防止一个子串在字符串中出现多次）。最后查一下就可。

Code

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=1e7+7,maxn1=1e3+5,mo=1e9+3;
char s[maxn1];
ll h[maxn][2],h1[maxn1],a[maxn1],b[maxn1];
ll n,m,p,i,t,j,k,l,x,y,ans;
void hash1(ll x){
ll t=x%maxn1;
while (h1[t] && h1[t]!=x) t=(t+1)%maxn1;
h1[t]=x;
}
void hash(ll x){
ll t=x%maxn;
while (h[t][0] && h[t][0]!=x) t=(t+1)%maxn;
h[t][0]=x;h[t][1]++;
}
int main(){
//  freopen("data.in","r",stdin);
scanf("%lld%lld%lld\n",&n,&m,&p);
b[0]=1;
for (i=1;i<maxn1;i++)
b[i]=b[i-1]*27%mo;
for (i=1;i<=n;i++){
memset(h1,0,sizeof(h1));
scanf("%s\n",s+1);t=strlen(s+1);
for (j=1;j<=t;j++)
a[j]=(a[j-1]*27+s[j]-96)%mo;
for (j=p;j<=t;j++){
x=(a[j]-a[j-p]*b[p]%mo+mo)%mo;
hash1(x);
}
for (j=0;j<maxn1;j++)
if (h1[j]) hash(h1[j]);
}
for (j=0;j<maxn;j++)
if (h[j][1]>=m) ans++;
printf("%lld\n",ans);
}