描述
http://www.lydsy.com/JudgeOnline/problem.php?id=2434
给出\(n\)个字符串,\(m\)个询问,对于第\(i\)个询问,求第\(x_i\)个字符串在第\(y_i\)个字符串中出现了多少次.
分析
首先我们可以想到对于串\(x\)和串\(y\),如果\(x\)在\(y\)中出现过,那么\(x\)一定是(\(y\)的某个前缀)的后缀,如果我们用所有字符串建立一个AC自动机,那么对于这个前缀,沿着失配边走,一定能走到\(x\)串.
所以我们得到了一个很直观但是复杂度很高的算法:对于每一个询问,枚举\(y\)中的字符作为前缀的结尾,沿着失配边走,看是否能够走到\(x\)串,复杂度是\(O(mL^2)\).显然这种算法是要超时的,我们需要考虑更优的算法.
这里先介绍fail树这一概念.对于AC自动机上的点\(x\),我们连一条\(f[x]\to{x}\)的有向边,这样就形成了一棵fail树.
这时我们会发现,之前说的能够沿着失配边找到\(x\)串的\(y\)串的后缀,一定在\(x\)的子树当中.我们用dfs序把树上问题转化为区间问题,问题就转化为了在区间内找符合条件的点有多少个.我们把\(y\)的后缀标为\(1\),其他全部标为\(0\),然后处理所有有关\(y\)的询问,由于是区间问题,可以用树状数组维护.
这样的算法的复杂度是\(O(mlogn)\)的.
#include <bits/stdc++.h>
using namespace std;
inline int read(int &x){x=;int k=;char c;for(c=getchar();c<''||c>'';c=getchar())if(c=='-')k=-;for(;c>=''&&c<='';c=getchar())x=x*+c-'';return x*=k;} const int maxn=1e5+,type=;
int n,qct;
int ans[maxn];
char s[maxn];
struct query{
int x[maxn],next[maxn],hd[maxn];
inline void add_query(int a,int b){
x[++qct]=a; next[qct]=hd[b]; hd[b]=qct;
}
}Q;
struct Aho_Corasick{
int sz,cnt,ect;
int q[maxn],hd[maxn],f[maxn],fa[maxn],val[maxn],pos[maxn],l[maxn],r[maxn],c[maxn<<];
int ch[maxn][type];
struct edge{
int to,next;
edge(){}
edge(int to,int next):to(to),next(next){}
}g[maxn];
Aho_Corasick(){sz=cnt=;memset(ch[],,sizeof ch[]);}
inline int id(char c){return c-'a';}
inline int lowbit(int x){return x&-x;}
inline void add_edge(int u,int v){g[++ect]=edge(v,hd[u]);hd[u]=ect;}
inline void build_trie(){
int m=strlen(s+);
int u=;
for(int i=;i<=m;i++){
int c=id(s[i]);
if(c=='P'-'a'){val[u]=++cnt,pos[cnt]=u;continue;}
if(c=='B'-'a'){u=fa[u];continue;}
if(!ch[u][c]){
memset(ch[++sz],,sizeof ch[sz]);
ch[u][c]=sz;
fa[sz]=u;
}
u=ch[u][c];
}
}
inline void get_fail(){
int L=,R=;
for(int c=;c<type;c++){
int u=ch[][c];
if(u){f[u]=;add_edge(,u);q[++R]=u;}
}
while(L<=R){
int u=q[L++];
for(int c=;c<type;c++){
int t=ch[u][c];
if(!t){ch[u][c]=ch[f[u]][c];continue;}
int v=f[u];
f[t]=ch[v][c];
add_edge(f[t],t);
q[++R]=t;
}
}
}
void dfs(int x,int &t){
l[x]=++t;
for(int i=hd[x];i;i=g[i].next) dfs(g[i].to,t);
r[x]=++t;
}
inline void add(int x,int d){
int R=sz<<;
while(x<=R){
c[x]+=d;
x+=lowbit(x);
}
}
inline int sum(int x){
int ret=;
while(x>){
ret+=c[x];
x-=lowbit(x);
}
return ret;
}
inline void solve(){
int m=strlen(s+);
int u=;
for(int i=;i<=m;i++){
int c=id(s[i]);
if(c=='B'-'a'){add(l[u],-);u=fa[u];continue;}
if(c=='P'-'a'){
int y=val[u];
for(int j=Q.hd[y];j;j=Q.next[j]){
int x=Q.x[j];
ans[j]=sum(r[pos[x]])-sum(l[pos[x]]-);
}
continue;
}
u=ch[u][c];
add(l[u],);
}
}
}ac;
inline void init(){
scanf("%s",s+);
ac.build_trie();
read(n);
for(int i=,a,b;i<=n;i++){
read(a); read(b);
Q.add_query(a,b);
}
}
inline void solve(){
ac.get_fail();
int t=;
ac.dfs(,t);
ac.solve();
for(int i=;i<=n;i++) printf("%d\n",ans[i]);
}
int main(){
init();
solve();
return ;
}
2434: [Noi2011]阿狸的打字机
Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2296 Solved: 1298
[Submit][Status][Discuss]
Description
阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有26个小写英文字母和'B'、'P'两个字母。
经阿狸研究发现，这个打字机是这样工作的：
l 输入小写字母，打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。
l 按一下印有'B'的按键，打字机凹槽中最后一个字母会消失。
l 按一下印有'P'的按键，打字机会在纸上打印出凹槽中现有的所有字母并换行，但凹槽中的字母不会消失。
例如，阿狸输入aPaPBbP，纸上被打印的字符如下：
a
aa
ab
我们把纸上打印出来的字符串从1开始顺序编号，一直到n。打字机有一个非常有趣的功能，在打字机中暗藏一个带数字的小键盘，在小键盘上输入两个数(x,y)（其中1≤x,y≤n），打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。
阿狸发现了这个功能以后很兴奋，他想写个程序完成同样的功能，你能帮助他么？
Input
输入的第一行包含一个字符串，按阿狸的输入顺序给出所有阿狸输入的字符。
第二行包含一个整数m，表示询问个数。
接下来m行描述所有由小键盘输入的询问。其中第i行包含两个整数x, y，表示第i个询问为(x, y)。
Output
输出m行，其中第i行包含一个整数，表示第i个询问的答案。
Sample Input
3
1 2
1 3
2 3
Sample Output
1
0
HINT
1<=N<=10^5