洛谷 P4094 [HEOI2016/TJOI2016]字符串 后缀数组+二分+主席树

题目链接
后缀数组
题目分析:
sa[i] – 第i小的后缀的编号
rank[i] --编号为i的后缀排第几;
height[i] – 第i和第i-1的最长lcp最长公共前缀;

1.二分答案,答案肯定在0—min(b-a+1, d-c+1);二分答案mid;
2.s[c–d], 满足条件的子串一定在rank[c] 开始向前后扩展的一个区间[L, R]中,(因为和某一个串排名越接近,那么与这个串的lcp(最长公共前缀)就越大)。
3.这个区间满足的条件就是这个区间任意一点i的height[i] >= mid; 那么这个区间中两两后缀的lcp一定大于等于mid。
4如果在这个区间存在i,sa[i] 编号大于等于a,并且sa[i] + mid - 1 <= b,那么这个mid就是可行的,可以继续二分下去,

5,这个区间L-R使用主席树来维持:

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
#include<map>
using namespace std;
typedef long long lt;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=100010;
int n,m,Q;
int a,b,c,d;
int rak[maxn],sa[maxn],tp[maxn],tax[maxn];
int height[maxn],mi[maxn][20];
int lft[maxn<<5],rht[maxn<<5],rt[maxn<<5],sum[maxn<<5],cnt;
char ss[maxn];

void rsort()
{
    for(int i=0;i<=m;++i) tax[i]=0;
    for(int i=1;i<=n;++i) tax[rak[i]]++;
    for(int i=1;i<=m;++i) tax[i]+=tax[i-1];
    for(int i=n;i>=1;--i) sa[tax[rak[tp[i]]]--]=tp[i];
}

void ssort()
{
    m=127;
    for(int i=1;i<=n;++i)
    rak[i]=ss[i-1],tp[i]=i;
    
    rsort(); 
    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;
        
    	rsort(); 
    	swap(rak,tp);
    	rak[sa[1]]=p=1;
    	for(int i=2;i<=n;++i)
        rak[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i]+k]==tp[sa[i-1]+k])?p:++p;
        if(p>=n)break;
    	m=p;
    }
}

void getH()
{
    int k=0;
    for(int i=1;i<=n;++i)
    {
        if(k) k--;
        int j=sa[rak[i]-1];
        while(ss[i+k-1]==ss[j+k-1]) k++;
        height[rak[i]]=k;
    }
}

void RMQ()
{
    for(int i=1;i<=n;++i) mi[i][0]=height[i];
    for(int j=1;(1<<j)<=n;++j)
    for(int i=1;i+(1<<j)-1<=n;++i)
    mi[i][j]=min(mi[i][j-1],mi[i+(1<<j-1)][j-1]);
}

int qmin(int ll,int rr)
{
    int k=0;
    while((1<<k+1)<=rr-ll+1) k++;
    return min(mi[ll][k],mi[rr-(1<<k)+1][k]);
}

int update(int pre,int ll,int rr,int x)
{
    int tt=++cnt; sum[tt]=sum[pre]+1;
    lft[tt]=lft[pre]; rht[tt]=rht[pre];
    int mid=ll+rr>>1;
    if(ll<rr)
    {
        if(x<=mid) lft[tt]=update(lft[pre],ll,mid,x);
        else rht[tt]=update(rht[pre],mid+1,rr,x);
    }
    return tt;
}

int query(int u,int v,int ll,int rr,int s,int t)
{
    if(ll<=s&&t<=rr) return sum[v]-sum[u];
    int res=0,mid=s+t>>1;
    if(ll<=mid) res+=query(lft[u],lft[v],ll,rr,s,mid);
    if(rr>mid) res+=query(rht[u],rht[v],ll,rr,mid+1,t);
    return res;
}

int qpos1(int ll,int rr,int x)
{
    int L=ll,R=rr,res=rr;
    while(L<R){
        int mid=L+R>>1;
        if(qmin(mid+1,rr)>=x) R=mid,res=mid;
        else L=mid+1;
    }
    return res;
}

int qpos2(int ll,int rr,int x)
{
    int L=ll,R=rr,res=ll;
    while(L<R){
        int mid=L+R>>1;
        if(qmin(ll+1,mid)>=x) L=mid+1,res=mid;
        else R=mid;
    }
    return res;
}

int check(int x)
{	
    int L=qpos1(1,rak[c],x);
    int R=qpos2(rak[c],n,x);

    if(query(rt[L-1],rt[R],a,b+1-x,1,n)>0) return 1;
    else return 0;
}

int main()
{
    n=read();Q=read();
    scanf("%s",ss);
    
    ssort(); getH(); RMQ();
    for(int i=1;i<=n;++i)
    rt[i]=update(rt[i-1],1,n,sa[i]);
    
    while(Q--)
    {
    	a=read();b=read();c=read();d=read();
    	int L=1,R=min(b-a+1,d-c+1),ans=0;
    	while(L<R)
    	{
    		int mid=L+R>>1;
    		if(check(mid)) L=mid+1,ans=mid;
    		else R=mid;
        }
        if(L==R) if(check(L)) ans=L;
        printf("%d\n",ans);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值