【BZOJ 2434】[Noi2011]阿狸的打字机 Ac自动机+树状数组+dfs序

很简单的想法是，建造Ac自动机以后，如果一个单词节点的fail指针指向另外一个字符串的路径上，那么它对两者之间会造成贡献，基于这样的想法，我们需要做的就是如何快速的求这个贡献。考虑反向链接fail指针每个节点只有一个出边所以将会是一个树，根据刚才的性质，根节点会对他的所有子树中的节点造成答案的贡献，然后维护dfs序，就能快速知道他的子树区间。最后统计答案的时候离线处理。对于一个串，将Trie数上的路径上的点对应的dfs序++（用树状数组维护），然后扫一遍问题，此时他作为y，那么就查找所有问到它作为y的问题，查询x的子树的区间和就是答案（根据刚才说的一个根节点一定是被包含在所有子树节点中）至此问题解决

#include<cstdio>#include<cstring>#include<iostream>#include<vector>#define maxn 200020using namespace std;int ch[maxn][26],fail[maxn],cnt,cur[maxn*2],q[maxn*2],head[maxn],tot,size,val[maxn];int in[maxn*2],out[maxn*2],n,nu[maxn*2],dep[maxn*2],c[maxn*4],len,ff[maxn*2];bool vis[maxn*2][26];char ss[maxn*2],cc[maxn*2];struct node{int a,b,ans;}nod[maxn*2];struct edge{int v,next;}e[maxn*2];vector<int>g[maxn];void adde(int a,int b){e[tot].v=b,e[tot].next=head[a];head[a]=tot++;}void update(int x,int add){while(x<=cnt){c[x]+=add;x+=x&(-x);}}int query(int x){int ans=0;while(x>0){ans+=c[x];x-=x&(-x);}return ans;}void insert(){int rt=0,id=0;for(int y,j=0;j<len;j++){if(ss[j]=='P')val[rt]=++id,nu[id]=rt;else if(ss[j]=='B')rt=ff[rt];else{y=ss[j]-'a';if(!ch[rt][y])ch[rt][y]=++size;ff[ch[rt][y]]=rt;rt=ch[rt][y];}}}void build_ac(){int l=0,r=1;q[0]=0;while(l<r){int u=q[l++];for(int i=0;i<26;i++){if(ch[u][i])fail[ch[u][i]]=u==0?0:ch[fail[u]][i],q[r++]=ch[u][i];else ch[u][i]=u==0?0:ch[fail[u]][i],vis[u][i]=true;}}for(int i=1;i<=size;i++){adde(fail[i],i);}}void dfs(int u){in[u]=++cnt;for(int v,i=head[u];i!=-1;i=e[i].next){v=e[i].v;dfs(v);}out[u]=cnt;}void solve(){int rt=0;for(int j=0;j<len;j++){if(ss[j]=='P'){for(int i=0;i<g[val[rt]].size();i++){int y=nu[nod[g[val[rt]][i]].a];//对应的那一个串的节点标号 int ans=query(out[y])-query(in[y]-1);nod[g[val[rt]][i]].ans=ans;}}else if(ss[j]=='B')update(in[rt],-1),rt=ff[rt];else rt=ch[rt][ss[j]-'a'],update(in[rt],1);}}int main(){memset(head,-1,sizeof(head));scanf("%s",ss);len=strlen(ss);insert();build_ac();dfs(0);scanf("%d",&n);int ccnt=0;for(int i=1;i<=n;i++){ccnt++;scanf("%d%d",&nod[ccnt].a,&nod[ccnt].b);g[nod[ccnt].b].push_back(ccnt);}solve();for(int i=1;i<=ccnt;i++)printf("%d\n",nod[i].ans);return 0;}