two strings

A1484. two strings（罗干）

时间限制：1.0s   内存限制：256.0MB

【问题描述】
　　给定两个字符串A和B，有五种操作，操作1为在A串开头添加一个字符，操作2为在A串结尾添加一个字符，操作3为在B串开头添加一个字符，操作4为在B串结尾添加一个字符，操作5为询问当前的B串在当前A串中出现的次数。保证字符均为小写字母，且A、B串初始非空。
【输入格式】
　　第一行第二行分别为初始的A串和B串；
　　第三行一个整数m，表示操作的次数；
　　接下来m行，每行表示一个操作，每行第一个数为一个在1-5之间的数字，若其值不为5，则在数字后会有一个小写字母。
【输出格式】
　　对于每个询问，每行输出一个整数，表示B串在A串中出现的次数。
【样例输入】
　　ababc
　　a
　　7
　　5
　　4 b
　　5
　　3 a
　　1 a
　　5
　　5
【样例输出】
　　2
　　2
　　1
　　1
【数据规模】
　　10%的数据中，最终A串和B串长度之和小于等于200，操作数小于等于10。
　　30%的数据中，最终A串和B串长度之和小于等于2000，操作数小于等于1000。
　　100%的数据中，最终A串和B串长度之和小于等于200000，操作数小于等于200000。

分析：对最终的A串和B串用分隔符拼接，构建后缀数组，模拟这m次操作；

　　　可以发现对于在A串或B串添加字符，最多会影响一个位置的可行性；

　　　在后缀数组中查找字符串可以使用二分找到连续区间，询问区间的合法性可以使用树状数组实现；

代码：

　　

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+,inf=0x3f3f3f3f;
int n,m,k,t,d[maxn],tot;
void add(int x,int y){while(x<=tot)d[x]+=y,x+=x&(-x);}
int get(int x){int ret=;while(x)ret+=d[x],x-=x&(-x);return ret;}
struct query{int op,pos;}q[maxn];
struct node{int x,id,nt;}e[maxn];
char ch[maxn];
int cntA[maxn],cntB[maxn];
int sa[maxn],lev[maxn],height[maxn];
int A[maxn],B[maxn],tsa[maxn];
int mi[][maxn],p[maxn];
int ask(int l,int r)
{
    int x=p[r-l+];
    return min(mi[x][l],mi[x][r-(<<x)+]);
}
void solve(int n,int m)
{
    for (int i = ; i < m; i ++) cntA[i] = ;
    for (int i = ; i <= n; i ++) cntA[ch[i]] ++;
    for (int i = ; i < m; i ++) cntA[i] += cntA[i - ];
    for (int i = n; i; i --) sa[cntA[ch[i]] --] = i;
    lev[sa[]] = ;
    for (int i = ; i <= n; i ++)
    {
        lev[sa[i]] = lev[sa[i - ]];
        if (ch[sa[i]] != ch[sa[i - ]]) lev[sa[i]] ++;
    }
    for (int l = ; lev[sa[n]] < n; l <<= )
    {
        memset(cntA,,sizeof(cntA[])*(n+));
        memset(cntB,,sizeof(cntB[])*(n+));
        for (int i = ; i <= n; i ++)
        {
            cntA[A[i] = lev[i]] ++;
            cntB[B[i] = (i + l <= n) ? lev[i + l] : ] ++;
        }
        for (int i = ; i <= n; i ++) cntB[i] += cntB[i - ];
        for (int i = n; i; i --) tsa[cntB[B[i]] --] = i;
        for (int i = ; i <= n; i ++) cntA[i] += cntA[i - ];
        for (int i = n; i; i --) sa[cntA[A[tsa[i]]] --] = tsa[i];
        lev[sa[]] = ;
        for (int i = ; i <= n; i ++)
        {
            lev[sa[i]] = lev[sa[i - ]];
            if (A[sa[i]] != A[sa[i - ]] || B[sa[i]] != B[sa[i - ]]) lev[sa[i]] ++;
        }
    }
    for (int i = , j = ; i <= n; i ++)
    {
        if (j) j --;
        while (ch[i + j] == ch[sa[lev[i] - ] + j]) j ++;
        height[lev[i]] = j;
    }
}
int main() {
    int i,j;
    //freopen("in.txt","r",stdin);
    tot=;
    int ha=-,ta=-,hb=-,tb=-;
    scanf("%s",ch+);
    for(i=;ch[i];i++)
    {
        e[tot].x=ch[i]-'a';
        if(!~ha)ha=ta=tot;
        else e[ta].nt=tot,ta=tot;
        tot++;
    }
    scanf("%s",ch+);
    for(i=;ch[i];i++)
    {
        e[tot].x=ch[i]-'a';
        if(!~hb)hb=tb=tot;
        else e[tb].nt=tot,tb=tot;
        tot++;
    }
    //cout<<ha<<" "<<ta<<" "<<hb<<" "<<tb<<endl;
    scanf("%d",&m);
    for(i=;i<=m;i++)
    {
        int op;
        char str[];
        scanf("%d",&op);
        q[i].op=op;
        if(op==)continue;
        scanf("%s",str);
        e[tot].x=str[]-'a';
        e[tot].id=i;
        if(op==)
        {
            e[tot].nt=ha;
            ha=tot;
        }
        else if(op==)
        {
            e[ta].nt=tot;
            ta=tot;
        }
        else if(op==)
        {
            e[tot].nt=hb;
            hb=tot;
        }
        else if(op==)
        {
            e[tb].nt=tot;
            tb=tot;
        }
        tot++;
    }
    tot=;
    int sx,sy,ex,ey;
    sx=sy=-;
    for(i=ha;;i=e[i].nt)
    {
        ch[++tot]=e[i].x;
        if(e[i].id)q[e[i].id].pos=tot;
        else {if(!~sx)sx=tot;ex=tot;}
        if(i==ta)break;
    }
    ch[++tot]=;
    for(i=hb;;i=e[i].nt)
    {
        ch[++tot]=e[i].x;
        if(e[i].id)q[e[i].id].pos=tot;
        else {if(!~sy)sy=tot;ey=tot;}
        if(i==tb)break;
    }
    solve(tot,);
    for(i=;i<=tot;i++)p[i]=+p[i>>];
    for(i=;<<i<=tot;i++)
    {
        for(j=;j+(<<i)<=tot;j++)
        {
            if(!i)mi[i][j]=height[j];
            else mi[i][j]=min(mi[i-][j],mi[i-][j+(<<(i-))]);
        }
    }
    //cout<<sx<<" "<<ex<<" "<<sy<<" "<<ey<<endl;
    for(i=sx;i+ey-sy<=ex;i++)add(lev[i],);
    for(i=;i<=m;i++)
    {
        int op=q[i].op;
        if(op==)
        {
            sx--;
            if(sx+ey-sy<=ex)add(lev[sx],);
        }
        else if(op==)
        {
            ex++;
            if(sx+ey-sy<=ex)add(lev[ex-ey+sy],);
        }
        else if(op==)
        {
            sy--;
            if(ex-ey+sy+>=sx)add(lev[ex-ey+sy+],-);
        }
        else if(op==)
        {
            ey++;
            if(ex-ey+sy+>=sx)add(lev[ex-ey+sy+],-);
        }
        else
        {
            int pos=lev[sy],len=ey-sy+;
            int l=,r=pos-,pl=pos,pr=pos;
            while(l<=r)
            {
                int mid=l+r>>;
                if(ask(mid+,pos)>=len)pl=mid,r=mid-;
                else l=mid+;
            }
            l=pos+,r=tot;
            while(l<=r)
            {
                int mid=l+r>>;
                if(ask(pos+,mid)>=len)pr=mid,l=mid+;
                else r=mid-;
            }
            printf("%d\n",get(pr)-get(pl-));
        }
    }
    return ;
}