luogu 5561 [Celeste-B]Mirror Magic 后缀数组+RMQ+multiset

思路肯定是没有问题，但是不知道为啥一直 TLE 两个点~

#include <bits/stdc++.h>
#define N 2000006
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
multiset<int>id,len;
multiset<int>::iterator it;
int arr[N],st[N],length[N],idxx[N],n,m=120;
char str[N];
namespace SA
{
    int rk[N], tp[N], tax[N], sa[N], height[N], Log[N], dp[N][22];
    void qsort()
    {
        for(int i=0;i<=m;++i) tax[i]=0;
        for(int i=1;i<=n;++i) ++tax[rk[i]];
        for(int i=1;i<=m;++i) tax[i]+=tax[i-1];
        for(int i=n;i>=1;--i) sa[tax[rk[tp[i]]]--]=tp[i];
    }
    void construct()
    {
        for(int i=1;i<=n;++i) rk[i]=arr[i],tp[i]=i;
        qsort();
        for(int k=1;k<=n;k<<=1)
        {
            int p=0;
            for(int i=n-k+1;i<=n;++i) tp[++p]=i;
            for(int i=1;i<=n;++i) if(sa[i]>k) tp[++p]=sa[i]-k;
            qsort(), swap(tp,rk), rk[sa[1]]=p=1;
            for(int i=2;i<=n;++i)
                rk[sa[i]]=(tp[sa[i-1]]==tp[sa[i]]&&tp[sa[i-1]+k]==tp[sa[i]+k])?p:++p;
            if(p==n) break;
            m=p;
        }
        int k=0;
        for(int i=1;i<=n;++i) rk[sa[i]]=i;
        for(int i=1;i<=n;++i)
        {
            if(k)--k;
            int j=sa[rk[i]-1];
            while(arr[i+k]==arr[j+k])++k;
            height[rk[i]]=k;
        }
    }
    void RMQ()
    {
        for(int i=2;i<=n;++i) Log[i]=Log[i>>1]+1;
        for(int i=1;i<=n;++i) dp[i][0]=height[i];
        for(int j=1;(1<<j)<=n;++j)
            for(int i=1;i+(1<<j)-1<=n;++i)
                dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
    }
    inline int getmin(int a,int b)
    {
        int k=Log[b-a+1];
        return min(dp[a][k],dp[b-(1<<k)+1][k]);
    }
};
int main()
{
    int i,j,pp,nn,q;
    // setIO("input");
    scanf("%d",&pp);
    for(i=1;i<=pp;++i)
    {
        scanf("%s",str+1);
        nn=strlen(str+1);
        st[i]=n+1;
        for(j=1;j<=nn;++j) arr[++n]=str[j]-'a'+1;
        arr[++n]=30;
    }
    SA::construct();
    SA::RMQ();
    scanf("%d",&q);
    for(i=1;i<=q;++i)
    {
        int opt;
        scanf("%d",&opt);
        if(opt==1)
        {
            int x,l,r,tmp;
            scanf("%d%d%d",&x,&l,&r); 

            tmp=st[x]+l-1;
            idxx[i]=SA::rk[tmp];
            length[i]=r-l+1;

            id.insert(idxx[i]);
            len.insert(r-l+1);                

            int a=(*id.begin()),b=(*(--id.end()));   

            if(a==b)
            {
                printf("%d\n",*len.begin());
            }
            else
            {
                printf("%d\n",min(*len.begin(),SA::getmin(a+1,b)));
            }
        }
        else
        {
            int k;
            scanf("%d",&k);     

            len.erase(len.lower_bound(length[k]));
            id.erase(id.lower_bound(idxx[k]));         

            if(len.empty())
            {
                printf("0\n");
                continue;
            }

            int a=(*id.begin()),b=(*(--id.end()));     

            if(a==b)
            {
                printf("%d\n",*len.begin());
            }
            else
            {
                printf("%d\n",min(*len.begin(),SA::getmin(a+1,b)));
            }
        }
    }
    return 0;
}