【后缀数组+阈值】CF741E Arpa’s abnormal DNA and Mehrdad’s deep interest

【题目】
CF
给定两个字符串 S , T S,T S,T,定义 R i R_i Ri表示在 S S S的第 i i i个和第 i + 1 i+1 i+1个字符之间插入 T T T得到的字符串( R 0 R_0 R0即加在开头, R n R_n Rn即加在结尾)。

Q Q Q个询问形如 ( l , r , k , x , y ) (l,r,k,x,y) (l,r,k,x,y),表示询问 l ≤ i ≤ r , x ≤ ( i m o d    k ) ≤ y l\leq i\leq r,x\leq (i\mod k) \leq y lir,x(imodk)y中字典序最小的 R i R_i Ri是哪个 i i i

∣ S ∣ , ∣ T ∣ , Q ≤ 1 0 5 |S|,|T|,Q\leq 10^5 S,T,Q105

【解题思路】
首先考虑如何比较在两个位置插入 T T T后字符串的字典序大小,显然可以按 T T T的端点拆成五个部分,用后缀数组求 l c p lcp lcp可以简单得到这五个部分的大小关系。
接下来要解决询问的问题,限制有取模不太好处理,发现可以写成 x + ⌊ i k ⌋ × k ≤ i ≤ y + ⌊ i k ⌋ × k x+\lfloor \frac i k \rfloor \times k\leq i\leq y+\lfloor \frac i k \rfloor \times k x+ki×kiy+ki×k的形式,那么我们不妨考虑对 k k k阈值 S S S

k > S k>S k>S,可以直接暴力枚举 ⌊ i k ⌋ \lfloor \frac i k \rfloor ki,剩下的问题就是在一段区间中取一个最优值,这一部分可以通过 RMQ \text{RMQ} RMQ预处理做到 O ( 1 ) O(1) O(1)单次查询,因此一次询问的复杂度是 O ( n S ) O(\frac n S) O(Sn)的,最坏情况即 O ( n 2 S ) O(\frac {n^2} S) O(Sn2)

k ≤ S k\leq S kS,不妨将询问离线,我们可以通过枚举 k k k,再枚举 x ∈ [ 0 , k − 1 ] x\in [0,k-1] x[0,k1],对于 i ≡ x   m o d   k i\equiv x\ mod\ k ix mod k的所有 i i i建一个 RMQ \text{RMQ} RMQ,询问的时候就在 [ x , y ] [x,y] [x,y]内查询,这部分总的复杂度是 O ( S n log ⁡ n ) O(Sn\log n) O(Snlogn)的。

根据均值不等式可以发现取 S = n log ⁡ n S=\sqrt {\frac n {\log n}} S=lognn 可以有最优复杂度 O ( n n log ⁡ n ) O(n\sqrt {n\log n}) O(nnlogn )

实际实现中,由于第二部分的复杂度与 RMQ \text{RMQ} RMQ长度相关,实际复杂度小于理论复杂度,而第一部分的复杂度与 k > S k>S k>S的询问次数相关,同样小于理论复杂度。在卡满的情况下应该 S S S取大一点会更优秀。

实现时建议使用 namespace \text{namespace} namespace避免 RMQ \text{RMQ} RMQ冲突。(当然你直接全部用一个数组来做 RMQ \text{RMQ} RMQ也是可以的)

【参考代码】

#include<bits/stdc++.h>
using namespace std;

const int N=2e5+10,lim=90,INF=0x3f3f3f3f;
int n,m,len,Q,fc[20],Log[N];

namespace IO
{
	int read()
	{
		int ret=0;char c=getchar();
		while(!isdigit(c)) c=getchar();
		while(isdigit(c)) ret=ret*10+(c^48),c=getchar();
		return ret;
	}
	void write(int x){if(x<0)x=-x,putchar('-');if(x>9)write(x/10);putchar(x%10^48);}
	void writesp(int x){write(x);putchar(' ');}
}
using namespace IO;

namespace SA
{
	int sa[N],rk[N],hi[N],h[18][N];
	int wx[N],wy[N],wa[N],wb[N];
	bool cmp(int *r,int a,int b,int l){return r[a]==r[b] && r[a+l]==r[b+l];}
	void getsa(int *r,int n,int m)
	{
		int *x=wa,*y=wb,*t,i,j,p;
		for(i=0;i<m;++i) wx[i]=0;
		for(i=0;i<n;++i) wx[x[i]=r[i]]++;
		for(i=1;i<m;++i) wx[i]+=wx[i-1];
		for(i=n-1;~i;--i) sa[--wx[x[i]]]=i;
		for(p=0,j=1;p<n;j<<=1,m=p)
		{
			for(p=0,i=n-j;i<n;++i) y[p++]=i;
			for(i=0;i<n;++i) if(sa[i]>=j) y[p++]=sa[i]-j;
			for(i=0;i<m;++i) wx[i]=0;
			for(i=0;i<n;++i) wx[wy[i]=x[y[i]]]++;
			for(i=1;i<m;++i) wx[i]+=wx[i-1];
			for(i=n-1;~i;--i) sa[--wx[wy[i]]]=y[i];
			for(t=x,x=y,y=t,i=1,p=1,x[sa[0]]=0;i<n;++i)
				x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
		}
	}
	void getheight(int *r,int n)
	{
		int i,j,k=0;
		for(i=1;i<=n;++i) rk[sa[i]]=i;
		for(i=0;i<n;hi[rk[i++]]=k)
			for(k?--k:0,j=sa[rk[i]-1];r[i+k]==r[j+k];++k);
	}
	void adjust(int n)
	{
		for(int i=n;i;--i) sa[i]++,rk[i]=rk[i-1];
		sa[0]=rk[0]=0;
	}
	void output(int n)
	{
		for(int i=0;i<=n;++i) printf("%d ",sa[i]); puts("");
		for(int i=0;i<=n;++i) printf("%d ",rk[i]); puts("");
		for(int i=0;i<=n;++i) printf("%d ",hi[i]); puts("");
	}
	void initrmq(int n)
	{
		for(int i=1;i<=n;++i) h[0][i]=hi[i];
		for(int j=1;j<18;++j) for(int i=1;i+fc[j]-1<=n;++i)
			h[j][i]=min(h[j-1][i],h[j-1][i+fc[j-1]]);
	}
	void buildsa(int *r,int n,int m)
	{
		getsa(r,n+1,m);getheight(r,n);adjust(n);initrmq(n);//output(n);
	}
	int calc(int x,int y)
	{
		int t=Log[y-x+1];
		//printf("calc:%d %d %d\n",x,y,min(h[t][x],h[t][y-fc[t]+1]));
		return min(h[t][x],h[t][y-fc[t]+1]);
	}
	int lcp(int x,int y)
	{
		x=rk[x];y=rk[y];
		if(x>y) swap(x,y);
		return calc(x+1,y);
	}
	int comp(int x,int y,int l)
	{
		if(lcp(x,y)>=l) return 0;
		return (rk[x]<rk[y])?-1:1;
	}
	bool cmpst(int x,int y)
	{
		int fg=x>y,t; if(fg) swap(x,y);
		if(m+x<=y)
		{
			t=comp(n+1,x+1,m);
			if(t) return (t<0)^fg;
			t=comp(x+1,x+m+1,y-(x+m));
			if(t) return (t<0)^fg;
			t=comp(y-m+1,n+1,m);
			if(t) return (t<0)^fg;
		}
		else
		{
			t=comp(n+1,x+1,y-x);
			if(t) return (t<0)^fg;
			t=comp(n+y-x+1,n+1,x+m-y);
			if(t) return (t<0)^fg;
			t=comp(x+1,m+n+x-y+1,y-x);
			if(t) return (t<0)^fg;
		}
		return fg^1;
	}
}

namespace DreamLolita
{
	int a[N],b[N],c[N],ans[N],mi[18][N];
	char s1[N],s2[N];

	struct data
	{
		int l,r,k,x,y,id;
		void in(int t){l=read();r=read();k=read();x=read();y=read();id=t;}
		bool operator <(const data&rhs){return k<rhs.k;}
	}q[N];
	void init()
	{
		scanf("%s%s",s1,s2);Q=read();
		n=strlen(s1);m=strlen(s2);len=n+m;
		for(int i=2;i<=len;++i) Log[i]=Log[i>>1]+1;
		fc[0]=1;for(int i=1;i<18;++i)fc[i]=fc[i-1]<<1;
		for(int i=0;i<n;++i)a[i]=s1[i]-'a'+1;
		for(int i=0;i<m;++i) a[n+i]=s2[i]-'a'+1;	
	}
	int calp(int x,int y)
	{
		if(x>y) return INF;
		int t=Log[y-x+1];
		//printf("calp:%d %d %d\n",x,y,min(mi[t][x],mi[t][y-fc[t]+1]));
		return min(mi[t][x],mi[t][y-fc[t]+1]);
	}
	void solvesmall()
	{
		sort(q+1,q+Q+1);
		for(int i=1,j=1;i<=lim && j<=Q;++i) if(i==q[j].k)//solve1
		{
			int t=j;
			while(t<Q && q[t+1].k==i) ++t;
			for(int k=0;k<i;++k)
			{
				int tmp=(n+1)/i+(k<=n%i);
				for(int i1=1;i1<=tmp;++i1) mi[0][i1]=c[(i1-1)*i+k];
				for(int j1=1;j1<18;++j1) for(int i1=1;i1+fc[j1]-1<=tmp;++i1)
					mi[j1][i1]=min(mi[j1-1][i1],mi[j1-1][i1+fc[j1-1]]);
				for(int i1=j;i1<=t;++i1)
				{
					//cerr<<q[i1].id<<" "<<ans[q[i1].id]<<endl;
					if(q[i1].x<=k && k<=q[i1].y) 
						ans[q[i1].id]=min(ans[q[i1].id],calp((q[i1].l-k+i-1)/i+1,(q[i1].r-k+i)/i));
					//cerr<<q[i1].id<<" "<<ans[q[i1].id]<<endl;
				}
			}
			j=t+1;
		}
	}
	void solvebig()
	{
		for(int i=1;i<=n+1;++i) mi[0][i]=c[i-1];
		for(int j=1;j<18;++j) for(int i=1;i+fc[j]-1<=n+1;++i)
			mi[j][i]=min(mi[j-1][i],mi[j-1][i+fc[j-1]]);
		for(int i=1;i<=Q;++i) if(q[i].k>lim)
		{
			for(int j=0;j*q[i].k<=n;++j) 
				ans[q[i].id]=min(ans[q[i].id],calp(max(j*q[i].k+q[i].x,q[i].l)+1,min(j*q[i].k+q[i].y,q[i].r)+1));
		}
	}
	void solve()
	{
		SA::buildsa(a,len,30);
		for(int i=0;i<=n;++i) b[i]=i;
		sort(b,b+n+1,SA::cmpst);
		//for(int i=0;i<=n;++i) printf("%d ",b[i]); puts("");
		for(int i=0;i<=n;++i) c[b[i]]=i;
		for(int i=1;i<=Q;++i) q[i].in(i),ans[i]=INF;
		solvesmall();solvebig();
		for(int i=1;i<=Q;++i) writesp(ans[i]<INF?b[ans[i]]:-1);
	}
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("CF741E.in","r",stdin);
	freopen("CF741E.out","w",stdout);
#endif
	DreamLolita::init();DreamLolita::solve();
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值