首页 > 代码库 > HDU 4821 String

HDU 4821 String

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



HDU 4821 String