链接
题解
广义后缀自动机自动机做法:
两个串建广义后缀自动机,然后 统计 个数,然后乘起来
后缀自动机做法:
加一个分隔符,塞进 ,然后还是 ,和上面说的那个流程一样
后缀数组做法(口胡,没写):
单调栈
代码(广义SAM)
#include <bits/stdc++.h>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<<x<<endl
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
ll c, f(1);
for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
for(;isdigit(c);c=getchar())x=x*10+c-0x30;
return f*x;
}
struct EXSAM
{
int tot, ch[maxn<<1][26], fa[maxn<<1], len[maxn<<1], pref[maxn<<1];
int* operator[](int u){return ch[u];}
void init()
{
int i;
rep(i,tot)cl(ch[i]),fa[i]=len[i]=pref[i]=0;
tot=1;
}
int append(int las, int c, int tag)
{
int p(las);
len[las=++tot]=len[p]+1;
pref[las]=tag;
for(;p and !ch[p][c];p=fa[p])ch[p][c]=las;
if(!p)fa[las]=1;
else
{
int q=ch[p][c];
if(len[q]==len[p]+1)fa[las]=q;
else
{
int qq=++tot;
memcpy(ch[qq],ch[q],sizeof(ch[q]));
fa[qq]=fa[q];
len[qq]=len[p]+1;
fa[q]=fa[las]=qq;
for(;ch[p][c]==q;p=fa[p])ch[p][c]=qq;
}
}
return las;
}
}exsam;
ll dp1[maxn<<1], dp2[maxn<<1], deg[maxn<<1];
char s[maxn], t[maxn];
void dp()
{
queue<int> q;
int i;
rep(i,exsam.tot)if(deg[i]==0)q.em(i);
while(!q.empty())
{
auto u=q.front(); q.pop();
if(exsam.pref[u]&1)dp1[u]++;
if(exsam.pref[u]&2)dp2[u]++;
auto f=exsam.fa[u];
dp1[f]+=dp1[u];
dp2[f]+=dp2[u];
if(--deg[f]==0)q.em(f);
}
}
int main()
{
ll i, n, m, lcp=0;
scanf("%s%s",s+1,t+1);
n=strlen(s+1), m=strlen(t+1);
rep(i,n)if(s[i]==t[i])lcp++;else break;
exsam.init();
ll p=1, p1, p2;
rep(i,lcp)p=exsam.append(p,s[i]-'a',3);
p1=p2=p;
for(i=lcp+1;i<=n;i++)p1=exsam.append(p1,s[i]-'a',1);
for(i=lcp+1;i<=m;i++)p2=exsam.append(p2,t[i]-'a',2);
rep(i,exsam.tot)deg[exsam.fa[i]]++;
ll ans=0;
dp();
rep(i,exsam.tot)ans+=dp1[i]*dp2[i]*(exsam.len[i]-exsam.len[ exsam.fa[i] ]);
printf("%lld",ans);
return 0;
}
代码(SAM)
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define maxe 1000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<<x<<endl
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
ll c, f(1);
for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
for(;isdigit(c);c=getchar())x=x*10+c-0x30;
return f*x;
}
struct SAM
{
int tot, las, ch[maxn<<1][27], fa[maxn<<1], len[maxn<<1], pref[maxn<<1];
int* operator[](int u){return ch[u];}
void init()
{
int i;
rep(i,tot)cl(ch[i]),fa[i]=len[i]=pref[i]=0;
tot=las=1;
}
void append(int c, int tag)
{
int p(las);
len[las=++tot]=len[p]+1;
pref[las]=tag;
for(;p and !ch[p][c];p=fa[p])ch[p][c]=las;
if(!p)fa[las]=1;
else
{
int q=ch[p][c];
if(len[q]==len[p]+1)fa[las]=q;
else
{
int qq=++tot;
memcpy(ch[qq],ch[q],sizeof(ch[q]));
fa[qq]=fa[q];
len[qq]=len[p]+1;
fa[q]=fa[las]=qq;
for(;ch[p][c]==q;p=fa[p])ch[p][c]=qq;
}
}
}
int mov(int p, int c){return max(1,ch[p][c]);}
}sam;
ll dp1[maxn<<1], dp2[maxn<<1], deg[maxn<<1];
char s[maxn], t[maxn];
void dp()
{
queue<int> q;
int i;
rep(i,sam.tot)if(deg[i]==0)q.em(i);
while(!q.empty())
{
auto u=q.front(); q.pop();
if(sam.pref[u]==1)dp1[u]++;
if(sam.pref[u]==2)dp2[u]++;
auto f=sam.fa[u];
dp1[f]+=dp1[u];
dp2[f]+=dp2[u];
if(--deg[f]==0)q.em(f);
}
}
int main()
{
ll i, n, m;
scanf("%s%s",s+1,t+1);
n=strlen(s+1), m=strlen(t+1);
sam.init();
rep(i,n)sam.append(s[i]-'a',1);
sam.append(26,0);
rep(i,m)sam.append(t[i]-'a',2);
rep(i,sam.tot)deg[sam.fa[i]]++;
ll ans=0;
dp();
rep(i,sam.tot)ans+=dp1[i]*dp2[i]*(sam.len[i]-sam.len[ sam.fa[i] ]);
printf("%lld",ans);
return 0;
}