题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4821
通过这一题回想起来字符串哈希的基本用法。
这道题很容易想到复杂度O(L*S/M)的算法,即把每次遍历的起点分为1-L,然后其中每次对字符串移动长度L,就可以省去很多重复判断。
详细的解释可以看看:https://www.cnblogs.com/acbingo/p/4719960.html。
一开始我直接用BKDRHASH来获得每次的hash值,但是由于这个操作是线性的,复杂度为O(M),直接T了。看了上面的文章,才想起以前学BKDRHASH时,可以获得O(1)的查询办法。
#include<cstring>
#include<iostream>
#include<string>
#include<cstdio>
#include<map>
using namespace std;
typedef unsigned long long ull;
const int maxn=1e5+5;
const ull base=31;
char s[maxn];
ull val[maxn],nbase[maxn];
inline void init(char* s){
int len=strlen(s);
for(int i=1;i<=len;i++)
val[i]=val[i-1]*base+s[i]-'a';
nbase[0]=1;
for(int i=1;i<=len;i++)
nbase[i]=nbase[i-1]*base;
}
inline ull getHash(int st,int ed){
return val[ed]-val[st-1]*nbase[ed-st+1];
}
map<ull,int>hav;
int main(void){
//freopen("in.txt","r",stdin);
int m,l; s[0]=1;
while (~scanf("%d %d",&m,&l)){
scanf("%s",s+1);
init(s);
int ans=0,len=strlen(s+1);
for(int i=1;i<=l&&i+m*l-1<=len;i++){
hav.clear();
for(int j=i;j+l<=i+m*l;j+=l){
ull res=getHash(j,j+l-1);
hav[res]++;
}
if(hav.size()==m)ans++;
for(int j=i+m*l;j+l-1<=len;j+=l){
ull res=getHash(j-m*l,j-m*l+l-1);
hav[res]--;
if(hav[res]==0)hav.erase(res);
res=getHash(j,j+l-1);
hav[res]++;
if(hav.size()==m)ans++;
}
}
printf("%d\n",ans);
}
return 0;
}