bzoj4566 [Haoi2016]找相同字符（SAM）

我们拿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;
}