Oulipo 【HDU - 1686】【哈希 | KMP】

题目链接

求模式串在待匹配串的出现次数。

Input

第一行是一个数字T，表明测试数据组数。
之后每组数据都有两行：第一行为模式串，长度不大于10000；第二行为待匹配串，长度不大于1000000。所有字符串只由大写字母组成。

Output

每组数据输出一行结果。

 

直接上哈希就是了——当然，这里用到的是KMP优化过后的哈希处理。

细节是，不要用取mod，直接用无符号类型，不然会被卡。

完整代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const ull hash1=131, hash2=233;
const int maxN=1e6+5;
char a[maxN], b[maxN];
ull a_hash1, b_hash1, ba1, a_hash2, b_hash2, ba2;
int main()
{
    int T;  scanf("%d", &T);
    while(T--)
    {
        getchar();
        scanf("%s", a);
        getchar();
        scanf("%s", b);
        int lena=(int)strlen(a), lenb=(int)strlen(b);
        a_hash1=b_hash1=a_hash2=b_hash2=0;
        ba1=ba2=1;
        for(int i=0; i<lena; i++)
        {
            a_hash1 = (a_hash1*hash1 + a[i]);
            a_hash2 = (a_hash2*hash2 + a[i]);
            b_hash1 = (b_hash1*hash1 + b[i]);
            b_hash2 = (b_hash2*hash2 + b[i]);
            ba1 = ba1*hash1;
            ba2 = ba2*hash2;
        }
        int ans=( ( a_hash1==b_hash1 && a_hash2==b_hash2 )?1:0 );
        for(int i=0; i+lena<lenb; i++)
        {
            b_hash1 = (b_hash1*hash1 + b[i+lena] - b[i]*ba1 );
            b_hash2 = (b_hash2*hash2 + b[i+lena] - b[i]*ba2 );
            if(b_hash1==a_hash1 && a_hash2==b_hash2) ans++;
        }
        printf("%d\n", ans);
    }
    return 0;
}