我们拿A串建出SAM,基数排序处理出拓扑序,算好|Right|。
用B串在SAM上跑,匹配到一个点p时计算当前匹配答案就是(len-mn[p]+1)*ri[p]。当然答案中还应该加上(mx[p]-mn[p]+1)*ri[p],对于所有的p在par树上的祖先,但是我们不能每次都直接跳父亲,这样复杂度会爆炸,因此我们只对p节点统计出现次数+1,然后处理p节点对答案的贡献。
我们最后再拓扑序一遍,把剩余节点对答案的贡献算出来。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
int n=0,m,son[N<<1][26],par[N<<1],mx[N<<1],sz[N<<1],rt,last,cnt[N<<1],a[N<<1],f[N<<1];
char s[N];ll ans=0;
inline void ins(int ch){
int p=last,np=++n;last=np;mx[np]=mx[p]+1;sz[np]=1;
for(;p&&!son[p][ch];p=par[p]) son[p][ch]=np;
if(!p){par[np]=rt;return;}
int q=son[p][ch];
if(mx[q]==mx[p]+1){par[np]=q;return;}
int nq=++n;memcpy(son[nq],son[q],sizeof(son[q]));mx[nq]=mx[p]+1;
par[nq]=par[q];par[q]=par[np]=nq;
for(;p&&son[p][ch]==q;p=par[p]) son[p][ch]=nq;
}
inline void gao(){
int p=rt,len=0;
for(int i=1;i<=m;++i){
if(son[p][s[i]-'a']){p=son[p][s[i]-'a'];++len;}
else{
while(p&&!son[p][s[i]-'a']) p=par[p];
if(!p) p=rt,len=0;
else len=mx[p]+1,p=son[p][s[i]-'a'];
}ans+=(ll)(len-mx[par[p]])*sz[p];cnt[p]++;
}
}
int main(){
// freopen("a.in","r",stdin);
scanf("%s",s+1);m=strlen(s+1);last=rt=++n;
for(int i=1;i<=m;++i) ins(s[i]-'a');
for(int i=1;i<=n;++i) cnt[mx[i]]++;
for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;--i) a[cnt[mx[i]]--]=i;
for(int i=n;i>=1;--i){int p=a[i];sz[par[p]]+=sz[p];}
memset(cnt,0,sizeof(cnt));scanf("%s",s+1);m=strlen(s+1);gao();
for(int i=n;i>=1;--i){
int p=a[i];f[par[p]]+=f[p]+cnt[p];if(p==rt) continue;
ans+=(ll)f[p]*sz[p]*(mx[p]-mx[par[p]]);
}printf("%lld\n",ans);
return 0;
}