HDU 4821 String(字符串哈希)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4821

题意：

给出M和L，和一个字符串S。要求找出S的子串中长度为L*M，并且可以分成M段，每段长L，并且M段都不相同的子串个数。哈希过后，
若Hash[i]-Hash[i+L]*K[L] == Hash[j]-Hash[j+L]*k[L]， 则说明串s[i...i+L-1]
和串s[j....j+L-1]是相同的。

#include<stdio.h>
#include<string.h>
#include<map>
#include<algorithm>
using namespace std;
typedef unsigned long long ULL;
#define SEED 31//这个数一般为素数 如1331, 1000000007等

const int maxn = 1e5 + 100;
ULL Hash[maxn], K[maxn];
map<ULL, int> st;
char s[maxn];
int main()
{
int M, L, len;
while(~scanf("%d%d", &M, &L))
{
scanf("%s", s);
len = strlen(s);
Hash[len] = 0;
K[0] = 1;
for(int i = 1; i <= L; i++)
K[i] = K[i-1] * SEED;
for(int i = len-1; i >= 0; i--)
{
Hash[i] = Hash[i+1]*SEED + (s[i]-'a'+1);
}
int res = 0;
for(int i = 0; i<L && i+M*L<len; i++)
{
st.clear();
for(int j = i; j < M*L+i; j+=L)
{
st[Hash[j]-Hash[j+L]*K[L]]++;
}
if(st.size() == M)
res++;
for(int j = M*L+i; j <= len-L; j+=L)
{
int head = j - M*L;
st[Hash[head]-Hash[head+L]*K[L]]--;
if(st[Hash[head]-Hash[head+L]*K[L]] == 0)
st.erase(Hash[head]-Hash[head+L]*K[L]);
st[Hash[j]-Hash[j+L]*K[L]]++;
if(st.size() == M)
res++;
}
}
printf("%d\n", res);
}
return 0;
}