好题啊……orz PoPoQQQ 大爷
一道相似的题目:【BZOJ】【3172】【TJOI2013】单词
那道题也是在fail树上数有多少个点,只不过这题是在x的fail树上数有多少个y的点。
感觉好难搞啊……那么我们不妨反过来……离线做?
既然是fail树!那么就看可以有dfs序,那么我们可以先找到整棵树上所有y的点,再看有哪些是在x的fail树上的,怎么做?x的fail树,对应的是整个dfs序上的一个区间!其实就是令dfs序上字符串y的点为1,其他为0,求一个区间和!用BIT就可以搞啦~
算法的大体框架就是:建AC自动机,搞出fail树的dfs序,在Trie树上dfs(枚举串y),将当前经过的这条链上的所有点在dfs序中对应的位置 置为1,如果走到某个字符串的结束点,一并处理所有与这个串相关的询问(对一个序列,分别查询多个区间和)。
TLE:一开始依旧写的以前的AC模板……其实那个是Trie图的,因为要出边补全,所以无论是否有这条出边都要找一遍fail,而这题只是一个AC自动机,不需要出边补全,所以可以大量减少找fail的复杂度……就轻松过了……
/************************************************************** Problem: 2434 User: Tunix Language: C++ Result: Accepted Time:440 ms Memory:18716 kb ****************************************************************/ //BZOJ 2434 #include<vector> #include<string> #include<map> #include<queue> #include<cstdio> #include<cstring> #include<cstdlib> #include<iostream> #include<algorithm> #define rep(i,n) for(int i=0;i<n;++i) #define F(i,j,n) for(int i=j;i<=n;++i) #define D(i,j,n) for(int i=j;i>=n;--i) #define pb push_back using namespace std; inline int getint(){ int v=0,sign=1; char ch=getchar(); while(ch<'0'||ch>'9'){ if (ch=='-') sign=-1; ch=getchar();} while(ch>='0'&&ch<='9'){ v=v*10+ch-'0'; ch=getchar();} return v*sign; } const int N=1e5+10,INF=~0u>>2; typedef long long LL; /******************tamplate*********************/ int to[N],nxt[N],head[N],cnt; void add(int x,int y){ to[++cnt]=y; nxt[cnt]=head[x]; head[x]=cnt; } int n,m,tot=1,num,pos[N]; struct node{ int ch[26],fail,sign,fa; }t[N]; char s1[N]; inline int id(char ch){return ch-'a';} queue<int>Q; void make_fail(){ Q.push(1); while(!Q.empty()){ int x=Q.front(),j,y; Q.pop(); add(t[x].fail,x); rep(i,26){ if (!t[x].ch[i]) continue; j=t[x].fail; while(j && !t[j].ch[i]) j=t[j].fail; y=t[x].ch[i]; t[y].fail=j ? t[j].ch[i] : 1; Q.push(y); } } } struct ques{int x,num;}; vector<ques>G[N]; int st[N],ed[N],c[N],dfs_clock,ans[N]; void ad(int x,int v){ for(int i=x;i<=tot;i+=i&(-i)) c[i]+=v; } int sum(int x){ if (x==0) return 0; int r=0; for(int i=x;i;i-=i&(-i)) r+=c[i]; return r; } void dfs(int x){ st[x]=++dfs_clock; for(int i=head[x];i;i=nxt[i]) dfs(to[i]); ed[x]=dfs_clock; } int main(){ #ifndef ONLINE_JUDGE freopen("2434.in","r",stdin); freopen("2434.out","w",stdout); #endif scanf("%s",s1); int l=strlen(s1); //build trie int x=1; rep(i,l){ if (s1[i]=='P'){ t[x].sign=++num; pos[num]=x; } else if (s1[i]=='B') x=t[x].fa; else{ int y=id(s1[i]); if (!t[x].ch[y]) t[x].ch[y]=++tot,t[tot].fa=x; x=t[x].ch[y]; } } make_fail(); //end build int T=getint(); F(i,1,T){ int x=getint(),y=getint(); G[y].pb((ques){x,i}); } dfs(1); x=1; rep(i,l){ if (s1[i]=='P'){ int y=t[x].sign; rep(j,G[y].size()){ int l=st[pos[G[y][j].x]],r=ed[pos[G[y][j].x]]; ans[G[y][j].num]=sum(r)-sum(l-1); } }else if (s1[i]=='B'){ ad(st[x],-1); x=t[x].fa; }else{ x=t[x].ch[id(s1[i])]; ad(st[x],1); } } F(i,1,T) printf("%d\n",ans[i]); return 0; }