[POJ 2774] Long Long Message （字符串哈希+二分）

链接

POJ 2774

---

题意

给出两个字符串，寻找其最长公共子串。
注意字符串的长度较大，100000。

---

思路

字符串哈希典型问题，寻找N个字符串的最长公共子串。对于N个长度不超过L的字符串，哈希可以在O(N*L*log(L))内解决此问题。本题N为2。
做法是对长度进行二分，对每个长度len，可以在线性时间内求出字符串s1中所有长度为len的哈希值，把这些哈希值存在哈希表中。同样对s2求出其长度为len的哈希值，对每个哈希值查找哈希表，如果哈希表中已存在该哈希值，则表示s1中含有相同子串。

---

代码

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cstdlib>
using namespace std;
typedef unsigned long long ulint;
//ulint seed = 100000007uLL;
ulint seed = 31uLL;
#define maxn 299999
char s1[maxn], s2[maxn];
int len1, len2;
int  bg[maxn], nx[maxn];
ulint H[maxn], xp[maxn];
bool check(int len)
{
    //printf("check len = %d\n", len);
    memset(bg, 0, sizeof(bg));

    ulint t = 1, ah = 0, bh = 0;
    int e = 0;
    for(int i = 0; i < len; i++)
    {
        t *= seed;
        ah = ah * seed + s1[i];
        bh = bh * seed + s2[i];
    }

    H[++e] = ah;
    nx[e] = bg[ah % maxn];
    bg[ah % maxn] = e;

    for(int i = len; i < len1; i++)
    {
        ah = ah * seed - s1[i-len] * t + s1[i];

        H[++e] = ah;
        nx[e] = bg[ah % maxn];
        bg[ah % maxn] = e;
    }

    for(int pos = bg[bh % maxn]; pos; pos = nx[pos])
    {
        if(H[pos] == bh) return true;
    }

    for(int i = len; i < len2; i++)
    {
        bh = bh * seed - s2[i-len] * t + s2[i];

        for(int pos = bg[bh % maxn]; pos; pos = nx[pos])
        {
            if(H[pos] == bh) return true;
        }
    }
    return false;
}

int main()
{
    //freopen("2774.txt", "r", stdin);

    scanf("%s%s", &s1, &s2);
    len1 = strlen(s1);
    len2 = strlen(s2);
    for(int i = 0; i < max(len1, len2); i++)
    {
        if(i < len1) s1[i] = s1[i] - 'a' + 1;
        if(i < len2) s2[i] = s2[i] - 'a' + 1;
    }

    int l = 1, r = min(len1, len2), m, res = 0;
    while(l <= r)
    {
        m = (l + r) >> 1;

        if(check(m))
        {
            //printf("ok\n");
            l = m + 1;
            res = max(res, m);
        }
        else r = m - 1;
    }

    cout << res << endl;

    return 0;
}