String
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4821
题目大意:
给定一个字符串(最长10^5)。从中选一个长度为 m * l 的子串,要求该子串能拆分为m个长度为 l 的一一不同的串。问有多少种取法。
解题思路:
比赛的时候,没有将串按照模 l 之后分类,导致TLE。。
字符串的哈希为:a[i] = a[i+1] * seek + str[i] - 'a' + 1;
那么j到j+len之间的字符串的哈希值是:a[j] - a[j+L]*base[L];
然后用map。
代码:
/*
字符串哈希
*/
#define seek 31
#define LL long long
#define ULL unsigned long long
#define maxn 100011
using namespace std;
char str[maxn];
ULL base[maxn], hh[maxn];
map<ULL, int> mp;
int main () {
base[0] = 1;
int m, l, len;
for (int i = 1; i < maxn; i++) base[i] = base[i - 1] * seek;
while(~scanf("%d%d%s", &m, &l, &str)) {
len = strlen(str);
hh[len] = 0;
int ans = 0;
for (int i = len - 1; i >= 0; i--) hh[i] = hh[i + 1] * seek + str[i] - 'a';
for (int i = 0; i < l && i + m * l <= len; i++) {
mp.clear();
ULL tmp;
for (int j = i; j < i + m * l; j += l) {
tmp = hh[j] - hh[j + l] * base[l];
mp[tmp]++;
}
if (mp.size() == m) ans++;
for (int j = i + m * l; j + l <= len; j += l) {
tmp = hh[j - m * l] - hh[j - (m - 1) * l] * base[l];
mp[tmp]--;
if (!mp[tmp]) mp.erase(tmp);
tmp = hh[j] - hh[j + l] * base[l];
mp[tmp]++;
if (mp.size() == m) ans++;
}
}
printf("%d\n", ans);
}
return 0;
}