链接: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;
}