子串查找 哈希入门

问题 A: 【 哈希和哈希表】子串查找

时间限制: 1 Sec  内存限制: 128 MB
提交: 118  解决: 39
[提交] [状态] [讨论版] [命题人:admin]

题目描述

这是一道模板题。
给定一个字符串A和一个字符串B，求B在A中的出现次数。A和B中的字符均为英语大写字母或小写字母。
A中不同位置出现的B可重叠。

 

输入

输入共两行，分别是字符串A和字符串B。

 

输出

输出一个整数，表示B在A中的出现次数。

 

样例输入

复制样例数据

zyzyzyz
zyz

样例输出

3

 

提示

1≤A,B的长度≤106，A、B仅包含大小写字母。

 

[提交][状态]

很入门的哈希题，当然KMP也可以

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int maxn = 1e6 + 100;
const int temp = 31;
string a, b;
ll f[maxn];
int la,lb;
ll hasha[maxn];
void init() {
    f[1] = temp;
    for (int i = 2; i < maxn; i++)
        f[i] = f[i - 1] * temp;
}

ll hashh(string s) {
    ll ans=0;
    for(int i=lb-1; i>=0; i--)
        ans=ans*temp+s[i];
    return ans;
}
void has(string s){
    for(int i=la-1; i>=0; i--){
        hasha[i]=hasha[i+1]*temp+s[i];
    }
}
int main() {
    init();
    cin >> a >> b;
    la = a.length();
    lb = b.length();
    ll hb = hashh(b);
    has(a);
    int cnt=0;
    for(int i=0; i<=la-lb; i++){
        if(hb==hasha[i]-hasha[i+lb]*f[lb])
            cnt++;
    }
    cout<<cnt<<endl;
    return 0;
}