bzoj千题计划318：bzoj1396: 识别子串（后缀自动机 + 线段树）

https://www.lydsy.com/JudgeOnline/problem.php?id=1396

后缀自动机的parent树上，如果不是叶子节点，那么至少有两个子节点

而一个状态所代表子串的出现次数就是子树中叶子节点的个数

所以只有叶子节点 即 |Right|=1的状态 代表的子串 出现了1次

我们计算以每一个位置为子串右端点时，它对一些位置的贡献

枚举|Right|=1的状态s

令end=Right(s)

那么以end为子串右端点，长度在[1,Max(parent(s))]的子串至少还会在s的父节点表示的状态中出现

所以在以end为识别子串右端点时

位置[end-Max(parent(s)),end]的最短长度为Max(parent(s)+1

位置[end-Max(s)+1,end-Max(parent(s))]的最短长度为 end-i+1

用两棵线段树维护

一棵直接维护最小值

另一棵维护end+1的最小值，查询的时候将结果-i

#include<cstdio>
#include<cstring>
#include<algorithm>

#define N 100001

using namespace std;

char s[N];

int ch[N<<][],tot=;
int fa[N<<],len[N<<];
int siz[N<<];
int last=,p,q,np,nq;

int leaf[N];

int v[N],sa[N<<];

struct Segment
{
    int mx[N<<];
    int tag[N<<];

    void down(int k)
    {
        mx[k<<]=min(mx[k<<],tag[k]);
        mx[k<<|]=min(mx[k<<|],tag[k]);
        tag[k<<]=min(tag[k<<],tag[k]);
        tag[k<<|]=min(tag[k<<|],tag[k]);
        tag[k]=2e9;
    }

    void build(int k,int l,int r)
    {
        mx[k]=tag[k]=2e9;
        if(l==r) return;
        int mid=l+r>>;
        build(k<<,l,mid);
        build(k<<|,mid+,r);
    }

    void change(int k,int l,int r,int opl,int opr,int w)
    {
        if(l>=opl && r<=opr)
        {
            mx[k]=min(mx[k],w);
            tag[k]=min(tag[k],w);
            return;
        }
        int mid=l+r>>;
        if(tag[k]!=2e9) down(k);
        if(opl<=mid) change(k<<,l,mid,opl,opr,w);
        if(opr>mid) change(k<<|,mid+,r,opl,opr,w);
        mx[k]=min(mx[k<<],mx[k<<|]);
    }

    int query(int k,int l,int r,int x)
    {
        if(l==r) return mx[k];
        int mid=l+r>>;
        if(tag[k]!=2e9) down(k);
        if(x<=mid) return query(k<<,l,mid,x);
        return query(k<<|,mid+,r,x);
    }
};

Segment tr1,tr2;

void extend(int c)
{
    len[np=++tot]=len[last]+;
    siz[tot]=;
    for(p=last;p && !ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=;
    else
    {
        q=ch[p][c];
        if(len[q]==len[p]+) fa[np]=q;
        else
        {
            nq=++tot;
            fa[nq]=fa[q];
            memcpy(ch[nq],ch[q],sizeof(ch[nq]));
            fa[q]=fa[np]=nq;
            len[nq]=len[p]+;
            for(;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
        }
    }
    last=np;
}

int main()
{
    scanf("%s",s+);
    int n=strlen(s+);
    for(int i=;i<=n;++i)
    {
        leaf[i]=tot+;
        extend(s[i]-'a');
    }
    for(int i=;i<=tot;++i) v[len[i]]++;
    for(int i=;i<=n;++i) v[i]+=v[i-];
    for(int i=;i<=tot;++i) sa[v[len[i]]--]=i;
    int x;
    for(int i=tot;i;--i)
    {
        x=sa[i];
        siz[fa[x]]+=siz[x];
    }
    tr1.build(,,n);
    tr2.build(,,n);
    int l,r,end;
    for(int x=;x<=tot;++x)
    if(siz[x]==)
    {
        l=len[fa[x]];
        r=len[x];
        end=len[x];
        tr1.change(,,n,end-l,end,l+);
        tr2.change(,,n,end-r+,end-l,end+);
    }
    int a,b;
    for(int i=;i<=n;++i)
    {
        a=tr1.query(,,n,i);
        b=tr2.query(,,n,i)-i;
        printf("%d\n",min(a,b));
    }
    return ;
}