P2414 [NOI2011]阿狸的打字机
AC自动机+树状数组
优质题解 <------题目分析
先AC自动机搞出Trie图
然后根据fail指针建一只新树
把树映射（拍扁）到一个序列上，用树状数组加速优化
在新树上处理时间戳，用于树状数组维护
在原Trie树上跑dfs查询答案。
因为Trie下标从0开始，树状数组从1开始
所以所有有关的下标都要注意
attention：树状数组上限一定要+1！（就是它卡了我1天TAT）
#include<iostream> #include<cstdio> #include<cstring> #include<queue> using namespace std; inline int max(int &a,int &b) {return a>b ?a:b;} const int N=1e5+2; struct Trie{int nxt_[26],nxt[26],fail,end,fa;}a[N]; int n,tot,cnt,word[N],ans[N]; struct AC_automaton{ //AC自动机 char q[N]; void Trie_build(){ scanf("%s",q); int u=0,len=strlen(q),st=0; while(q[st]=='B'||q[st]=='P') ++st; for(int i=st;i<len;++i){ if(q[i]=='B') u=a[u].fa; //删除一位 else if(q[i]=='P') word[++tot]=u,a[u].end=tot; //加入新单词 else{ int p=q[i]-'a'; if(!a[u].nxt[p]) a[u].nxt[p]=++cnt,a[cnt].fa=u; u=a[u].nxt[p]; } } } void AC_build(){ queue <int> h; for(int i=0;i<26;++i) if(a[0].nxt[i]) h.push(a[0].nxt[i]); while(!h.empty()){ int x=h.front(); h.pop(); for(int i=0;i<26;++i){ int &to=a[x].nxt[i]; if(to){ a[to].fail=a[a[x].fail].nxt[i]; h.push(to); }else to=a[a[x].fail].nxt[i]; } } } void backup(){ //对Trie树的nxt进行备份由于下面的遍历（AC自动机会改变nxt） for(int i=0;i<=cnt;++i) for(int j=0;j<26;++j) a[i].nxt_[j]=a[i].nxt[j]; } }mo1; struct tree_array{ //树状数组 int c[N]; inline void add(int x,int k) {for(;x<=cnt+1;x+=x&-x) c[x]+=k;} //上限要+1！ inline int sum(int x){int res=0; for(;x;x-=x&-x) res+=c[x]; return res;} }mo2; int cnt1,hd1[N],nxt1[N],ed1[N],poi1[N]; int cnt2,hd2[N],nxt2[N],ed2[N],poi2[N],id[N]; inline void add_edge1(int x,int y){ //fail树邻接表 nxt1[ed1[x]]=++cnt1; hd1[x]= hd1[x] ? hd1[x]:cnt1; ed1[x]=cnt1; poi1[cnt1]=y; } inline void add_edge2(int x,int y,int _id){ //存询问邻接表 nxt2[ed2[x]]=++cnt2; hd2[x]= hd2[x] ? hd2[x]:cnt2; ed2[x]=cnt2; poi2[cnt2]=y; id[cnt2]=_id; } struct new_tree{ int dfs_clock,dfn[N],low[N]; inline void dfs1(int x){ //fail树遍历 dfn[x]=++dfs_clock; //时间戳 for(int i=hd1[x];i;i=nxt1[i]) dfs1(poi1[i]); low[x]=dfs_clock; //size[x]=low[x]-dfn[x] } inline void dfs2(int x){ //Trie树遍历 mo2.add(dfn[x],1); //保证只有该条路径上 if(a[x].end) //该点是某个单词的结尾 { for(int i=hd2[a[x].end];i;i=nxt2[i]){ int e=word[poi2[i]]; ans[id[i]]=mo2.sum(low[e])-mo2.sum(dfn[e]-1); } } for(int i=0;i<26;++i) if(a[x].nxt_[i]) dfs2(a[x].nxt_[i]); //沿原树遍历 mo2.add(dfn[x],-1); } }mo3; int main(){ mo1.Trie_build(); mo1.backup(); mo1.AC_build(); for(int i=1;i<=cnt;++i) add_edge1(a[i].fail,i); //fail树连边 scanf("%d",&n); int q1,q2; for(int i=1;i<=n;++i) scanf("%d%d",&q1,&q2),add_edge2(q2,q1,i); //把询问存到邻接表上 mo3.dfs1(0); mo3.dfs2(0); for(int i=1;i<=n;++i) printf("%d\n",ans[i]); return 0; }
