HDU - 4080 Stammering Aliens(字符串hash + 二分)

点击打开题目链接

题目大意：

给出一个m和一个字符串s，寻找s的最长子串满足：该子串在s中出现不小于m次。如果有多个子串满足，选择最右侧开始的子串。要求输出子串的出现次数和最右侧一个子串的出现位置。

思路：

容易想到用BKDR哈希函数和二分子串长度来做。
WA了两次，第一次solved函数里ans的初始值设置错了，不应该设置为0，改成-1；第二次二分结果后不应该直接输出t，因为t有-1情况，转成了ans。

代码：

#include<bits/stdc++.h>

using namespace std;
const int maxn = 40000 + 5;
const int seed = 31;
typedef unsigned long long ull;
ull _base[maxn], _hash[maxn];
int len;
map<ull, ull> mp;
int m;
string s;

ull str_hash(int l, int r) {
    return _hash[l] - _hash[r+1] * _base[r-l+1];
}

int solved(int l) {
    mp.clear();
    int ans = -1;
    for(int i = 0; i <= len - l; i++) {
        ull t = str_hash(i, i+l-1);
        mp[t]++;
        if(mp[t] >= m) ans = i;
    }
    return ans;
}

int main() {
    _base[0] = 1;
    for(int i = 1; i < maxn; i++) _base[i] = _base[i-1] * seed;
    while(cin >> m, m) {
        cin >> s;
        len = s.length();
        _hash[len] = 0;
        for(int i = len-1; i >= 0; i--) _hash[i] = _hash[i+1] * seed + (s[i] - 'a' + 1);
        int l = 1, r = len - m + 1, t, ans;
        if(solved(l) == -1) {cout << "none" << endl; continue;}
        while(l <= r) {
            int mid = (l+r) >> 1;
            t = solved(mid);
            if(t != -1) {
                l = mid + 1;
                ans = t;
            }
            else r = mid - 1;
        }
        cout << r << ' ' << ans << endl;
    }
    return 0;
}