bzoj1396识别子串（SAM+线段树）

复习SAM板子啦！考前刷水有益身心健康当然这不是板子题/水题……

很容易发现只在i位置出现的串一定是个前缀串。那么对答案的贡献分成两部分：一部分是len[x]-fa~len[x]的这部分贡献会是r-l+1；剩下一部分1~len-fa-1这部分会和i~r构成答案，写两棵线段树即可。

然后就又是板子题了，两个板子（SAM+线段树）套起来。

#include<bits/stdc++.h>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
const int N=2e5+,inf=0x3f3f3f3f;
int n,lst=,rt=,cnt=,ch[N][],len[N],fa[N],sz[N];
char s[N];
struct tree{
    int mn[N<<];
    void init(){memset(mn,0x3f,sizeof mn);}
    void pushdown(int rt)
    {
        if(mn[rt]==inf)return;
        mn[rt<<]=min(mn[rt<<],mn[rt]),mn[rt<<|]=min(mn[rt<<|],mn[rt]);
        mn[rt]=inf;
    }
    void update(int L,int R,int v,int l,int r,int rt)
    {
        if(L<=l&&r<=R){mn[rt]=min(mn[rt],v);return;}
        int mid=l+r>>;pushdown(rt);
        if(L<=mid)update(L,R,v,lson);
        if(R>mid)update(L,R,v,rson);
    }
    int query(int k,int l,int r,int rt)
    {
        if(l==r)return mn[rt];
        int mid=l+r>>;pushdown(rt);
        if(k<=mid)return query(k,lson);
        return query(k,rson);
    }
}tr1,tr2;
void build(int c)
{
    int p=lst,np=++cnt;
    len[np]=len[p]+,sz[np]=;
    while(p&&!ch[p][c])ch[p][c]=np,p=fa[p];
    if(!p)fa[np]=rt;
    else{
        int q=ch[p][c];
        if(len[p]+==len[q])fa[np]=q;
        else{
            int nq=++cnt;
            memcpy(ch[nq],ch[q],sizeof ch[q]);
            fa[nq]=fa[q],fa[np]=fa[q]=nq,len[nq]=len[p]+;
            while(p&&ch[p][c]==q)ch[p][c]=nq,p=fa[p];
        }
    }
    lst=np;
}
int main()
{
    scanf("%s",s+),n=strlen(s+);
    for(int i=;i<=n;i++)build(s[i]-'a');
    for(int i=;i<=cnt;i++)sz[i]=;
    for(int i=;i<=cnt;i++)sz[fa[i]]=;
    tr1.init(),tr2.init();
    for(int i=;i<=cnt;i++)
    if(sz[i]==)
    {
        int l=len[i]-len[fa[i]],r=len[i];
        if(l>=)tr1.update(,l-,r+,,n,);
        tr2.update(l,r,r-l+,,n,);
    }
    for(int i=;i<=n;i++)printf("%d\n",min(tr1.query(i,,n,)-i,tr2.query(i,,n,)));
}