suffix automaton

struct state
{
    int len,nex[27],pre;
    state(int len=0,int pre=-1):
        len(len),pre(pre)
    {
        memset(nex,0,sizeof(nex));
    }
} sam[2000005];

int suffam(char *s)
{
    int lenth=strlen(s+1);
    int idx=0,last=0;
    for(int i=1; i<=lenth; i++)
    {
        int now=++idx;
        int t=s[i]-'a'+1;
        sam[now].len=i;
        while(~last && !sam[last].nex[t])
        {
            sam[last].nex[t]=now;
            last=sam[last].pre;
        }
        if(last==-1)
            sam[now].pre=0;
        else
        {
            int nex=sam[last].nex[t];
            if(sam[last].len+1==sam[nex].len)
                sam[now].pre=nex;
            else
            {
                sam[++idx]=state(sam[last].len+1,sam[nex].pre);
                for(int i=1;i<=26;i++)
                    sam[idx].nex[i]=sam[nex].nex[i];
                sam[now].pre=sam[nex].pre=idx;
                while(~last && sam[last].nex[t]==nex)
                    sam[last].nex[t]=idx,last=sam[last].pre;
            }
        }
        last=now;
    }
    return idx;
}

你可能感兴趣的:(suffix automaton)