HDU4270 Dynamic Lover（后缀自动机）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4270

题意：给出一个串s，有三种操作：（1）将一个串t接在s之后；（2）给出len，输出不大于len的后缀和长度为len的子串中字典序最小的；（3）给出len，删掉长度为len的后缀。

思路：建立s的后缀自动机。（1）直接将t插入；（2）从last向前，标记所有的终结点。然后从head开始DFS；（3）将所有删去的节点以及以复制得到的节点标记删去，方法是共同指向一个标记。

const int KIND=26;
struct SAM
{
    SAM *son[KIND],*pre;
    int len,cnt,flag,id,*isDel;

    void init()
    {
        clr(son,NULL);
        pre=NULL;
        id=flag=len=0;
    }
};

SAM sam[N],*head,*last;
int cnt,len;
char s[N];
int isDel[N],delCnt,posMap[N];

void initSAM()
{
    sam[0].init();
    head=last=&sam[0];
    isDel[delCnt]=0;
    head->isDel=&isDel[delCnt++];
    cnt=1;
}


void insert(int x)
{
    SAM *p=&sam[cnt],*u=last;
    p->init();
    p->len=p->id=last->len+1;
    posMap[p->len]=cnt++;

    isDel[delCnt]=0;
    p->isDel=&isDel[delCnt++];
    last=p;
    
    for(;u&&(!u->son[x]||*u->son[x]->isDel);u=u->pre) u->son[x]=p;
    if(!u) p->pre=head;
    else if(u->son[x]->len==u->len+1) p->pre=u->son[x];
    else
    {
        SAM *r=&sam[cnt++],*q=u->son[x];
        *r=*q; r->len=u->len+1;
        p->pre=q->pre=r;
        for(;u&&u->son[x]==q;u=u->pre) u->son[x]=r;
    }
}


void insert()
{
    RD(s);
    int L=strlen(s),i;
    FOR0(i,L) insert(s[i]-'a'),len++;
}

int dep,flag;

int DFS(SAM *p,int _dep)
{
    if(_dep==dep) return p->id-_dep+1;
    if(p->flag==flag) return len-_dep+1;
    int i;
    FOR0(i,26) if(p->son[i]&&!(*p->son[i]->isDel))
    {
        return DFS(p->son[i],_dep+1);
    }
}

void find()
{
    flag++;
    SAM *p=last;
    while(p&&p!=head) p->flag=flag,p=p->pre;
    head->flag=0;
    RD(dep);
    PR(DFS(head,0));
}



void del()
{
    int L;
    RD(L);
    int i;
    FOR(i,len-L+1,len) *sam[posMap[i]].isDel=1;
    len-=L;
    last=&sam[posMap[len]];
}

int main()
{
    while(scanf("%s",s)!=-1)
    {
        delCnt=0;
        initSAM(); len=strlen(s);
        int i;
        FOR0(i,len) insert(s[i]-'a');
        int Q,op;
        RD(Q);
        while(Q--)
        {
            RD(op);
            if(op==1) insert();
            else if(op==2) find();
            else del();
        }
    }
    return 0;
}