bzoj 4556 [Tjoi2016&Heoi2016]字符串 二分+后缀数组+主席树+RMQ

求一个后缀数组和height的rmq。
对rank建一个主席树。

二分第二个串的匹配长度len。

再二分出后缀数组中与第二个串lcp大于等于len的左右边界lq,rq。

设四个点为l1,r1,l2,r2。那么第一个串只有开头为[l1,r1-len+1]的后缀可以匹配。

那么如果rank[l1,r1-len+1]有在区间[lq,rq]中的,那么可行。
这个可以到主席树中查询。

时间复杂度O(nlog^2)。
辣鸡bz卡常数。一开始20s的时限强行改成10s并不知道为什么。。。

#include <bits/stdc++.h>
using namespace std;
#define N 110000
#define M 5100000
int n,m,l1,r1,l2,r2,lq,rq;
int bir[N],root[N],bit[N];
char s[N];
int h[21][N],sa[N],has[N],tr[N],rank[N];
int ch[M][2],num[M],cnt;
char getc()
{
    static const int LEN = 4096;
    static char buf[LEN],*S=buf,*T=buf;
    if(S == T)
    {
        T = (S=buf)+fread(buf,1,LEN,stdin);
        if(S == T)return EOF;
    }
    return *S++;
}
int read()
{
    static char ch;
    static int D;
    while(!isdigit(ch=getc()));
    for(D=ch-'0'; isdigit(ch=getc());)
        D=(D<<3)+(D<<1)+(ch-'0');
    return D;
}
void getstr(char *s)
{
    static char ch;
    while(!islower(ch=getc()));
    for(;islower(ch);ch=getc())
        *s=ch,s++;
}
int cmp(int x,int y,int k)
{
    if(x+k>n||y+k>n)return 0;
    return rank[x]==rank[y]&&rank[x+k]==rank[y+k];
}
void getsa()
{
    int i,cnt;
    for(i=1;i<=n;i++)has[s[i]-'a'+1]++;
    for(i=1,cnt=0;i<=26;i++)if(has[i])tr[i]=++cnt;
    for(i=1;i<=26;i++)has[i]+=has[i-1];
    for(i=1;i<=n;i++)rank[i]=tr[s[i]-'a'+1],sa[has[s[i]-'a'+1]--]=i;
    for(int k=1;cnt!=n;k<<=1)
    {
        for(i=1;i<=n;i++)has[i]=0;
        for(i=1;i<=n;i++)has[rank[i]]++;
        for(i=1;i<=n;i++)has[i]+=has[i-1];
        for(i=n;i>=1;i--)if(sa[i]>k)tr[sa[i]-k]=has[rank[sa[i]-k]]--;
        for(i=1;i<=k;i++)tr[n-i+1]=has[rank[n-i+1]]--;
        for(i=1;i<=n;i++)sa[tr[i]]=i;
        for(i=1,cnt=0;i<=n;i++)tr[sa[i]]=cmp(sa[i],sa[i-1],k) ? cnt:++cnt;
        for(i=1;i<=n;i++)rank[i]=tr[i];
    }
    for(int i=1;i<=n;i++)
    {
        if(rank[i]==1)continue;
        for(int j=max(h[0][rank[i-1]]-1,1);;j++)
        {
            if(s[i+j-1]==s[sa[rank[i]-1]+j-1])h[0][rank[i]]=j;
            else break;
        }
    }
    for(int i=1;i<=16;i++)
        for(int j=1;j<=n;j++)
            if(j-(1<<i)>=1)
                h[i][j]=min(h[i-1][j],h[i-1][j-(1<<i-1)]);
}
void insert(int l,int r,int pre,int &now,int pos)
{
    num[now=++cnt]=num[pre]+1;
    if(l==r)return;
    ch[now][0]=ch[pre][0];
    ch[now][1]=ch[pre][1];
    int mid=l+r>>1;
    if(mid>=pos)insert(l,mid,ch[pre][0],ch[now][0],pos);
    else insert(mid+1,r,ch[pre][1],ch[now][1],pos);
}
int query(int l,int r,int pre,int now)
{
    if(lq<=l&&r<=rq)
        return num[now]-num[pre];
    int mid=l+r>>1;
    if(mid>=rq)return query(l,mid,ch[pre][0],ch[now][0]); 
    if(mid<lq) return query(mid+1,r,ch[pre][1],ch[now][1]);
    return query(l,mid,ch[pre][0],ch[now][0])||query(mid+1,r,ch[pre][1],ch[now][1]);
}
int check(int x)
{
    int t=rank[l2],l=1,r=t-1;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(h[bir[t-mid]][t]<x||h[bir[t-mid]][mid+bit[t-mid]]<x)l=mid+1;
        else r=mid-1;
    }
    lq=l;l=t+1,r=n;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(h[bir[mid-t]][mid]<x||h[bir[mid-t]][t+bit[mid-t]]<x)r=mid-1;
        else l=mid+1;
    }
    rq=r;
    return query(1,n,root[l1-1],root[r1-x+1]);
}
int main()
{
    n=read();m=read();
    getstr(s+1);
    getsa();
    for(int i=1;i<=n;i++)
        insert(1,n,root[i-1],root[i],rank[i]);
    for(int i=1,j=0;i<=n;i++)
    {
        if((1<<j+1)<=i)j++;
        bir[i]=j;bit[i]=1<<j;
    }
    for(;m--;)
    {
        l1=read();r1=read();l2=read();r2=read();
        int l=1,r=min(r2-l2+1,r1-l1+1);
        while(l<=r)
        {
            int mid=l+r>>1;
            if(check(mid))l=mid+1;
            else r=mid-1;
        }
        printf("%d\n",r);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值