HDU 4821 String

来源:互联网 发布:自动建站源码 编辑:程序博客网 时间:2024/05/15 23:49

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 100011using 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;}



0 0
原创粉丝点击