Codeforces.666E.Forensic Examination(广义后缀自动机 线段树合并)

题目链接

\(Description\)

给定串\(S\)和\(m\)个串\(T_i\)。\(Q\)次询问,每次询问\(l,r,p_l,p_r\),求\(S[p_l\sim p_r]\)在\(T_l\sim T_r\)中的哪个串出现次数最多,输出最多次数以及它是\(T\)中的第几个。若最多的有多个,输出下标最小的。

\(Solution\)

挺好的题吧

对\(T\)个串建SAM,然后要求出SAM每个节点上\(|right|\)最大的是哪个串。

每个节点的\(|right|\)可以在DFS parent树时合并子节点得到。如果用线段树,区间\(|right|\)最大的是哪个串也可以维护出来。

那么可以离线,在每个点处处理该点上的询问,边DFS边合并线段树得到所有答案。(当然可持久化一下在线也行?)

怎么得到\(S[p_l\sim p_r]\)在SAM上的匹配节点呢?

维护一个节点指针\(p\),拿\(S\)在SAM上尽可能匹配,匹配不了就跳\(fa\)。这样能保证当前\(S\)的后缀(\(S[i]\))一定在\(p\)节点出现了。

所以在\(i\)这里处理\(p_r=i\)的询问。只要我们从当前节点\(p\)一直跳\(fa\),就能找到\(S[p_l,p_r]\)所在的节点,其答案就是该节点的\(|right|\)状态。可以用倍增实现。

(就是从\(S[1,p_r]\)所匹配的节点\(p\)往上跳,跳到从上往下第一个\(len_x\geq p_r-p_l+1\)的节点\(x\),\(x\)就是\(S[p_l,p_r]\)所在的节点)

如果出现次数为\(0\)的话也要输出最靠前的(即\(l\))。=-=

唉 6点多写完代码调到现在 心累

//529ms	56700KB
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define Bit 16
const int N=5e5+5,M=5e4+5,S=M<<1; int m,root[S],fa[S][18];
char s[N],tmp[M];
struct Edge
{
int Enum,H[N],nxt[N],to[N];
inline void AddEdge(int u,int v){
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
}
}Pos,Qy;
struct Edge2
{
int Enum,H[S],nxt[S],to[S];
inline void AddEdge(int u,int v){
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
}
}Par;
struct Queries{
int l,r,pl,pr;
}q[N];
struct Suffix_Automaton
{
int tot,las,fa[S],son[S][26],len[S]; Suffix_Automaton() {tot=las=1;}
void Insert(int c)
{
int np=++tot,p=las; len[las=np]=len[p]+1;
for(; p&&!son[p][c]; p=fa[p]) son[p][c]=np;
if(!p) fa[np]=1;
else
{
int q=son[p][c];
if(len[q]==len[p]+1) fa[np]=q;
else
{
int nq=++tot; len[nq]=len[p]+1;
memcpy(son[nq],son[q],sizeof son[q]);
fa[nq]=fa[q], fa[q]=fa[np]=nq;
for(; son[p][c]==q; p=fa[p]) son[p][c]=nq;
}
}
}
}sam;
struct Node
{
int val,id;
bool operator <(const Node &x)const{
return val<x.val||(val==x.val&&id>x.id);
}
}Ans[N];
struct Segment_Tree
{
#define S M*17
#define lson son[x][0]
#define rson son[x][1]
int tot,son[S][2];
Node node[S];
#undef S
#define Update(x) node[x]=std::max(node[lson],node[rson]);
void Insert(int &x,int l,int r,int pos)
{
x=++tot;
if(l==r) return (void)(node[x]=(Node){1,pos});
int m=l+r>>1;
pos<=m ? Insert(lson,l,m,pos) : Insert(rson,m+1,r,pos);
Update(x);//Update
}
int Merge(int x,int y)
{
if(!x||!y) return x^y;
if(!lson&&!rson) return node[x].val+=node[y].val, x;//叶节点,合并right
lson=Merge(lson,son[y][0]), rson=Merge(rson,son[y][1]);
Update(x); return x;
}
Node Query(int x,int l,int r,int L,int R)
{
if(!x) return (Node){0,L};
if(L<=l && r<=R) return node[x];
int m=l+r>>1;
if(L<=m)
if(m<R) return std::max(Query(lson,l,m,L,R),Query(rson,m+1,r,L,R));
else return Query(lson,l,m,L,R);
return Query(rson,m+1,r,L,R);
}
}T; inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-'0',c=gc());
return now;
}
void DFS(int x)
{
for(int i=Par.H[x]; i; i=Par.nxt[i])
DFS(Par.to[i]), root[x]=T.Merge(root[x],root[Par.to[i]]);
for(int i=Pos.H[x],id; i; i=Pos.nxt[i])
id=Pos.to[i], Ans[id]=T.Query(root[x],1,m,q[id].l,q[id].r);
} int main()
{
scanf("%s",s+1); int n=strlen(s+1);
m=read();
for(int i=1; i<=m; ++i)
{
scanf("%s",tmp), sam.las=1;
for(int j=0,l=strlen(tmp); j<l; ++j)
sam.Insert(tmp[j]-'a'), T.Insert(root[sam.las],1,m,i);//不就是每位的|right[las]|=1吗→_→我在纠结什么
//就是las节点的线段树上,i位置的|right|=1,给i位置+1
}
int Q=read();
for(int i=1; i<=Q; ++i)
q[i]=(Queries){read(),read(),read(),read()}, Qy.AddEdge(q[i].pr,i);
int lim=sam.tot;
for(int x=2; x<=lim; ++x) Par.AddEdge(fa[x][0]=sam.fa[x],x);
for(int i=1; i<=Bit; ++i)
for(int x=2; x<=lim; ++x)
fa[x][i]=fa[fa[x][i-1]][i-1];
for(int c,now=0,p=1,i=1; i<=n; ++i)
{
if(sam.son[p][c=s[i]-'a']) p=sam.son[p][c], ++now;//!!!靠 这写错调了两个小时 唉 心累
else
{
for(c=s[i]-'a'; p&&!sam.son[p][c]; p=sam.fa[p]);
if(!p) {p=1, now=0; continue;}
now=sam.len[p]+1, p=sam.son[p][c];
}
for(int j=Qy.H[i],len,id; j; j=Qy.nxt[j])
{
id=Qy.to[j];
if(now<(len=q[id].pr-q[id].pl+1)) continue;
int x=p;
for(int i=Bit; ~i; --i)
if(sam.len[fa[x][i]]>=len) x=fa[x][i];
Pos.AddEdge(x,id);
}
}
DFS(1);
for(int i=1; i<=Q; ++i)
if(!Ans[i].val) printf("%d 0\n",q[i].l);
else printf("%d %d\n",Ans[i].id,Ans[i].val); return 0;
}
上一篇:TFS 安装遇到的问题


下一篇:求职(2015南京站获得百度、美的集团、趋势科技、华为offer)