众所周知,武辰延很喜欢字符串。
这天,他对着两个字符串 s 和 t 发呆,他发现这两个串的前缀有很多相似的地方,s 的两个前缀连接起来竟也是 t 的前缀。
武辰延想知道有多少对 s 的非空前缀连接起来是 t 的前缀。
形式化地讲,我们把si 看作字符串 s 长度为 i 的前缀。
对于一对前缀 (si,sj) (允许 i=j)而言,当满足si+sj=ti+j 时,我们认为这两个 s 的前缀拼接后等于 t 的一个前缀。
两对 s 的前缀 (si,sj) 与 (si′,sj′) 不同当且仅当 i'j′ 。
思路:
因为在找到某个前缀匹配时,前缀的前缀也一定匹配,因此具有单调性,可以二分找。(算个trick吧)
剩下就是哈希了
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e5+100;
typedef long long LL;
const LL base=131;
const LL mod=998244353;
inline LL read(){LL x=0,f=1;char ch=getchar(); while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
char s1[maxn],s2[maxn];
LL sum1[maxn],sum2[maxn],p[maxn];
LL cal(LL l,LL r,LL sum[]){
return (sum[r]%mod-sum[l-1]*p[r-l+1]%mod+mod)%mod;
}
bool check(LL l1,LL r1,LL l2,LL r2,LL len2){
LL t1=cal(l1,r1,sum1);
LL t2=0;
if(r2>len2) t2=0;
else t2=cal(l2,r2,sum2);
if(t1==t2) return true;
else return false;
}
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
p[0]=1;
for(LL i=1;i<maxn;i++) p[i]=p[i-1]*base%mod;
cin>>(s1+1);LL len1=strlen(s1+1);
cin>>(s2+1);LL len2=strlen(s2+1);
for(LL i=1;i<=len1;i++) sum1[i]=(sum1[i-1]*base%mod+(s1[i]-'a'+1)%mod)%mod;///不能减-'a',会变0然后冲撞,以后字符串hash统一写s[i]就好
for(LL i=1;i<=len2;i++) sum2[i]=(sum2[i-1]*base%mod+(s2[i]-'a'+1)%mod)%mod;
LL ans=0;
for(LL i=1;i<=len2;i++){
if(s1[i]!=s2[i]) break;
if(s2[i+1]!=s1[1]) continue;///L卡到1的边界
LL l=1;LL r=len1;///二分第一个串的额外前缀长度为mid
while(l<r){
LL mid=(l+r+1)>>1;
if(check(1,mid,i+1,i+mid,len2)) l=mid;
else r=mid-1;
}
ans+=(l);
}
cout<<ans<<"\n";
return 0;
}