「2.1」子串查找

「2.1」子串查找

问题背景

Oulipo

题目描述

这是一道模板题。

给定一个字符串 A 和一个字符串 B，求 B 在 A 中的出现次数。A 和 B 中的字符均为英语大写字母或小写字母。

A 中不同位置出现的 B 可重叠。

输入格式

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

输出格式

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

样例输入1

zyzyzyz
zyz

样例输出1

3

注释说明

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

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ULL;
string a,b;
int la,lb,ans;
ULL ha,hb[1000005],s=1;
ULL geth(int y,int x) {
	return hb[x]-hb[y-1]*s;
}
int main() {
	cin>>b>>a;
	lb=b.size();
	la=a.size();
	b=" "+b;
	a=" "+a;
	for(int i=1; i<=la; i++) {
		ha=ha*131+a[i];
		s*=131;
	}//printf("%dd ",ha);
	for(int i=1; i<=lb; i++){
		hb[i]=hb[i-1]*131+b[i];
		//printf("%d \n",hb[i]);
	}
	hb[0]=0;
	for(int i=1; i<=lb-la+1; i++) {
		if(ha==hb[i+la-1]-hb[i-1]*s) {
			ans++;//printf("%d\n",i);
		}
	}
	cout<<ans<<endl;
}/*
#include<bits/stdc++.h>
using namespace std;
char a[1000005],b[1000005];
int main() {
    gets(a); gets(b);
    int n=strlen(a), m=strlen(b);
    long long hs=0, k=1, hb=0;
    for (int i=0; i<m; i++) hs=hs*97+a[i], hb=hb*97+b[i], k*=97;
    int ans=(hs==hb);  //前m个字符的子串如果匹配，ans++
    for (int i=m; i<n; i++) {
        hs=hs*97+a[i]-k*a[i-m];  //求枚举起点是i-m，终点是i的子串的hash值
        ans+=(hs==hb);  //如果匹配，ans++
    }
    printf("%d\n", ans);
}
*/