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