【BZOJ3879】SvT（后缀自动机，虚树）

【BZOJ3879】SvT（后缀自动机，虚树）

题面

BZOJ

题解

看着这个东西，询问若干个前缀两两之间的\(lcp\)？
显然\(lcp\)就是\(SAM\)构建出来的\(parent\)数上的\(LCA\)所代表的长度。
那么这样子就转为了树型\(dp\)。
然后发现是前缀？把串转过来就是后缀了。
\(\sum t\)是\(O(n)\)级别的？显然虚树。
那么直接虚树搞搞就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAX 1010000
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int n,m;char ch[MAX];
long long ans;
struct Node
{
	int son[26];
	int len,ff;
}t[MAX];
int last=1,tot=1,lt[MAX];
void extend(int c)
{
	int p=last,np=++tot;last=tot;
	t[np].len=t[p].len+1;
	while(p&&!t[p].son[c])t[p].son[c]=np,p=t[p].ff;
	if(!p)t[np].ff=1;
	else
	{
		int q=t[p].son[c];
		if(t[q].len==t[p].len+1)t[np].ff=q;
		else
		{
			int nq=++tot;
			t[nq]=t[q];t[nq].len=t[p].len+1;
			t[q].ff=t[np].ff=nq;
			while(p&&t[p].son[c]==q)t[p].son[c]=nq,p=t[p].ff;
		}
	}
}
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int dfn[MAX],low[MAX],dep[MAX],fa[MAX],tim,size[MAX],hson[MAX],top[MAX];
void dfs1(int u,int ff)
{
	size[u]=1;dep[u]=dep[ff]+1;fa[u]=ff;
	for(int i=h[u];i;i=e[i].next)
	{
		int v=e[i].v;
		dfs1(v,u);size[u]+=size[v];
		if(size[v]>size[hson[u]])hson[u]=v;
	}
}
void dfs2(int u,int tp)
{
	dfn[u]=++tim;top[u]=tp;
	if(hson[u])dfs2(hson[u],tp);
	for(int i=h[u];i;i=e[i].next)
		if(e[i].v!=hson[u])dfs2(e[i].v,e[i].v);
	low[u]=tim;
}
int LCA(int u,int v)
{
	while(top[u]^top[v])(dep[top[u]]<dep[top[v]])?v=fa[top[v]]:u=fa[top[u]];
	return dep[u]<dep[v]?u:v;
}
int p[MAX],S[MAX];
bool vis[MAX];
bool cmp(int a,int b){return dfn[a]<dfn[b];}
int dp(int u)
{
	int ret=0;if(vis[u])ret=1;
	for(int i=h[u];i;i=e[i].next)
	{
		int d=dp(e[i].v);
		ans+=1ll*t[u].len*d*ret;
		ret+=d;
	}
	return ret;
}
int main()
{
	n=read();m=read();
	scanf("%s",ch+1);reverse(&ch[1],&ch[n+1]);
	for(int i=1;i<=n;++i)extend(ch[i]-97),lt[i]=last;
	for(int i=1;i<=tot;++i)if(t[i].ff)Add(t[i].ff,i);
	dfs1(1,0);dfs2(1,1);memset(h,0,sizeof(h));cnt=1;
	while(m--)
	{
		int t=read(),top=0;cnt=0;
		for(int i=1;i<=t;++i)vis[p[i]=lt[n-read()+1]]=true;
		sort(&p[1],&p[t+1],cmp);
		for(int i=t;i>1;--i)p[++t]=LCA(p[i],p[i-1]);
		sort(&p[1],&p[t+1],cmp);t=unique(&p[1],&p[t+1])-p-1;
		for(int i=1;i<=t;++i)
		{
			while(top&&low[S[top]]<dfn[p[i]])--top;
			Add(S[top],p[i]);S[++top]=p[i];
		}
		ans=0;dp(p[1]);printf("%lld\n",ans);
		for(int i=1;i<=t;++i)h[p[i]]=0,vis[p[i]]=false;
	}
	return 0;
}