4556: [Tjoi2016&Heoi2016]字符串(后缀自动机做法)

题目大意:给出两段区间,求其中一段的所有子串与另一段的最长公共前缀。

做法:听说可以用后缀数组作,不过忘得差不多了,与是愉快的敲了sam。好像还挺有收获的。

首先原串反过来建,这样题目就转化成了一段的所有子串与另一段的最长公共后缀,这样就可以识别了。

假设要在s【a,b】中找最长后缀在s【c,d】中出现过(反串后),那么我们先找到以b结尾的前缀在parent树上的位置,

接着我们二分答案mid,然后在parent树上倍增,使其max值为大于等于mid的最小值。


这样就能使right集合尽量大,更可能在s【c,d】出现过。

因为一个节点的right集合等于它在parent树上的子树的并集,所以我们可以在每个节点上开一棵线段树,查询其right集合中是否存在(a+mid-1,b)中的数就可以判断了。


关于线段树合并:

因为每个点都要开线段树,自然要用动态开点的线段树。由于要与每个儿子都合并,所以一般的打法会挂掉。(可以认真想一下)

只要在每次合并开个新点就可以了。

int  merge(int x1,int x2)
{
	if(!x1||!x2) return x1+x2;
	int x=++trlen;
	tr[x].lc=merge(tr[x1].lc,tr[x2].lc);
	tr[x].rc=merge(tr[x1].rc,tr[x2].rc);
	return x;
}
code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
struct SAM{
	int par,max,a[27];
}sam[200010];int rt=1,tot=1,tail=1;
struct node{
	int x,y,next;
}a[200010];int last[200010],Len=0;
struct trnode{
	int lc,rc;
}tr[4000010];int trlen=0,root[200010];
int n,m,fa[200010][22],dep[200010],pre[100010];
char s[100010];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void update(int &x,int l,int r,int k)
{
	if(x==0) x=++trlen;
	if(l==r) return;
	int mid=(l+r)/2;
	if(k<=mid) update(tr[x].lc,l,mid,k);
	else update(tr[x].rc,mid+1,r,k);
}
int  merge(int x1,int x2)
{
	if(!x1||!x2) return x1+x2;
	int x=++trlen;
	tr[x].lc=merge(tr[x1].lc,tr[x2].lc);
	tr[x].rc=merge(tr[x1].rc,tr[x2].rc);
	return x;
}
bool findans(int x,int l,int r,int fl,int fr)
{
	if(x==0) return false;
	if(l==fl&&r==fr) return true;
	int mid=(l+r)/2,lc=tr[x].lc,rc=tr[x].rc;
	if(fr<=mid) return findans(lc,l,mid,fl,fr);
	if(fl>mid) return findans(rc,mid+1,r,fl,fr);
	return findans(lc,l,mid,fl,mid)||findans(rc,mid+1,r,mid+1,fr);
}
void addsam(int c,int len)
{
    int p=tail,np=++tot;
    sam[np].max=len;update(root[np],1,n,len);
    for(;p&&!sam[p].a[c];p=sam[p].par) sam[p].a[c]=np;
    tail=np;
    if(!p) sam[np].par=rt;
    else
    {
        int q=sam[p].a[c];
        if(sam[q].max==sam[p].max+1) sam[np].par=q;
        else
        {
            int nq=++tot;sam[nq]=sam[q];
            sam[nq].max=sam[p].max+1;
            sam[q].par=sam[np].par=nq;
            for(;p&&sam[p].a[c]==q;p=sam[p].par) sam[p].a[c]=nq;
        }
    }
}
void ins(int x,int y)
{
	a[++Len].x=x;a[Len].y=y;
	a[Len].next=last[x];last[x]=Len;
}
void dfs(int x,int f)
{
	dep[x]=dep[f]+1;fa[x][0]=f;
	for(int i=1;(1<<i)<=dep[x];i++)
		fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;dfs(y,x);
		root[x]=merge(root[x],root[y]);
	}
}
bool check(int len,int r,int fl,int fr)
{
	if(len==0) return true;
	int x=pre[r];
	if(fr-fl+1<len) return false;
	for(int i=20;i>=0;i--)
		if((1<<i)<=dep[x]&&sam[fa[x][i]].max>=len) x=fa[x][i];
	if(findans(root[x],1,n,fl+len-1,fr)) return true;
	return false;
}
int solve(int x,int y,int fl,int fr)
{
	int l=0,r=y-x+1,ans=0;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(check(mid,y,fl,fr)) ans=mid,l=mid+1;
		else r=mid-1;
	}
	return ans;
}
int main()
{
	scanf("%d %d",&n,&m);
	scanf("%s",s+1);
	for(int i=n;i>=1;i--) addsam(s[i]-'a',n-i+1);
	for(int i=2;i<=tot;i++) ins(sam[i].par,i);
	int x=rt;for(int i=1;i<=n;i++) x=sam[x].a[s[n-i+1]-'a'],pre[i]=x;
	dep[0]=-1;dfs(1,0);
	while(m--)
	{
		int A=read(),B=read(),C=read(),D=read();
		printf("%d\n",solve(n-D+1,n-C+1,n-B+1,n-A+1));
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值