做法:字符串哈希+二分
思路:
- 我们可以先枚举两个字符串相同前缀的长度
然后再枚举s前缀和t剩余后面部分有多少相同的部分,每次加上这一部分长度即可
这一部分可以采用二分这部分长度来优化
// Problem: 武辰延的字符串
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/9984/B
// Memory Limit: 524288 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp(aa,bb) make_pair(aa,bb)
#define _for(i,b) for(int i=(0);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,b,a) for(int i=(b);i>=(a);i--)
#define mst(abc,bca) memset(abc,bca,sizeof abc)
#define X first
#define Y second
#define lowbit(a) (a&(-a))
#define debug(a) cout<<#a<<":"<<a<<"\n"
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
typedef long double ld;
const int N=100010;
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-6;
const double PI=acos(-1.0);
string s,t;
const ull P = 131;
ull powP[N],H1[N],H2[N];
ll ans;
string str;
void init(int n){
powP[0]=1;
for(int i=1;i<=n;i++){
powP[i]=powP[i-1]*P;
}
}
void calH(ull H[],string str){
H[0]=0;
for(int i=1;i<str.size();i++){
H[i]=H[i-1]*P+str[i];
}
}
ull get1(int l,int r){
return H1[r]-H1[l-1]*powP[r-l+1];
}
ull get2(int l,int r){
return H2[r]-H2[l-1]*powP[r-l+1];
}
void solve(){
cin>>s>>t;
int len1=s.size(),len2=t.size();
s=" "+s;t=" "+t;
init(len2);
calH(H1,s);calH(H2,t);
for(int i=1;i<=min(len1,len2-1);i++){
if(s[i]==t[i]){
int lenl=1,lenr=len2-i,res=0;
while(lenl<=lenr){
int mid=(lenl+lenr)>>1;
int l1=1,r1=l1+mid-1,l2=i+1,r2=l2+mid-1;
if(get1(l1,r1)==get2(l2,r2)) res=mid,lenl=mid+1;
else lenr=mid-1;
}
ans+=res;
}
else break;
}
cout<<ans<<"\n";
}
int main(){
ios::sync_with_stdio(0);cin.tie(0);
// int t;cin>>t;while(t--)
solve();
return 0;
}