bzoj2434: [Noi2011]阿狸的打字机

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2434

题意：中文题。

分析：给定一个这样的字符串很容易就想建一颗trie树。然后在trie树上进行匹配，这不就是AC自动机么！我们可以直接考虑这样的一种暴力，当查询x,y时我们只要将从root到y路径上所有的点沿着fail跑一遍，只要经过一次x就ans+1。但是这样会超时，我们在观察这个结构的特征，然后会发现fail[i]是唯一的那么反向的话就是一颗树咯！那样就是在x的子树中找在路径root-y上的点的个数咯。子树中的统计问题我们很容易就能想到用dfs序来处理成区间问题，那root-y上的点呢？我们可以按照建trie树的过程去处理，当走到y时我们就将所有{x}-y的问题全部处理掉，我们只要在遇到小写字母时就将它的in[i]位置插入1，遇到B是就将当前位置减去1，那么当我们遇到y是就只剩下一条链上的in[i]的位置上有1了，那么答案就是区间[in[x],out[x]]中的1的个数咯，用过树状数组维护下就好了。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=100100;
const int MAX=100000000;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=998244353;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int tot,u[N],v[N],w[N],pre[N];
void add(int x,int y,int z) {
    v[tot]=y;w[tot]=z;pre[tot]=u[x];u[x]=tot++;
}
char s[N];
struct trie {
    int g,m,sz,X[N],fa[N],val[N],fail[N],tr[N][26];
    int ttot,tu[N],tv[N],tpre[N];
    int k,in[N],out[N];
    void trie_clear() {
        k=g=m=sz=ttot=0;
        memset(tu,-1,sizeof(tu));
    }
    void trie_insert(char *ch) {
        while (*ch!='\0') {
            if (*ch=='B') { m=fa[m];ch++;continue ; }
            if (*ch=='P') { val[m]=++g;X[g]=m;ch++;continue ; }
            if (tr[m][*ch-'a']) { m=tr[m][*ch-'a'];ch++; }
            else { fa[sz+1]=m;tr[m][*ch-'a']=++sz;m=sz;ch++; }
        }
    }
    void trie_build() {
        queue<int>Q;
        for (int i=0;i<26;i++)
        if (tr[0][i]) Q.push(tr[0][i]);
        while (!Q.empty()) {
            int now=Q.front();Q.pop();
            for (int i=0;i<26;i++)
            if (tr[now][i]) {
                fail[tr[now][i]]=tr[fail[now]][i];
                Q.push(tr[now][i]);
            } else tr[now][i]=tr[fail[now]][i];
        }
    }
    void add(int x,int y) {
        tv[ttot]=y;tpre[ttot]=tu[x];tu[x]=ttot++;
    }
    void add_b() {
        for (int i=0;i<=sz;i++)
        if (fail[i]!=i) add(fail[i],i);
    }
    void dfs(int x) {
        k++;in[x]=k;
        for (int i=tu[x];i!=-1;i=tpre[i]) dfs(tv[i]);
        out[x]=k;
    }
}ac;
int len,f[N],ans[N];
void updata(int x,int y) {
    for (;x<=len+1;x+=x&(-x)) f[x]+=y;
}
int getsum(int x) {
    int ret=0;
    for (;x;x-=x&(-x)) ret+=f[x];
    return ret;
}
int main()
{
    int i,j,n,x,y;
    scanf("%s", s);
    ac.trie_clear();
    ac.trie_insert(s);
    ac.trie_build();
    ac.add_b();ac.dfs(0);
    scanf("%d", &n);
    tot=0;memset(u,-1,sizeof(u));
    for (i=1;i<=n;i++) {
        scanf("%d%d", &x, &y);add(y,x,i);
    }
    len=strlen(s);x=0;
    memset(f,0,sizeof(f));
    for (i=0;i<len;i++) {
        if (s[i]=='B') {
            updata(ac.in[x],-1);x=ac.fa[x];continue ;
        }
        if (s[i]=='P') {
            for (j=u[ac.val[x]];j!=-1;j=pre[j]) {
                ans[w[j]]=getsum(ac.out[ac.X[v[j]]])-getsum(ac.in[ac.X[v[j]]]-1);
            }
            continue ;
        }
        x=ac.tr[x][s[i]-'a'];updata(ac.in[x],1);
    }
    for (i=1;i<=n;i++) printf("%d\n", ans[i]);
    return 0;
}