【BZOJ】2434: [Noi2011]阿狸的打字机

题意

给你一些字符串。\(m\)次询问,每一次询问第\(x\)个字符串在\(y\)字符串中出现了多少次。(输入总长$ \le 10^5$, \(M \le 10^5\))

分析

在ac自动机上,\(x\)字符串出现的所有位置就是其它节点的fail树上有这个节点的节点。即fail树中,\(x\)字符串终止节点的子树。

题解

根据分析,我们只要构造ac自动机和fail树,按dfs序依次便历ac自动机,将节点到根的路径打上标记。每到一个终止节点,就更新以这个节点为模板的匹配串的答案,即这些节点子树的和。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
char buf[20*N], *is=buf;
int ihead[N], cnt, last[N], c[N][26], fail[N], pos[N], FF[N], LL[N], tot, su[N], ans[N];
inline bool isdig(const char &c) {
return c>='0'&&c<='9';
}
inline bool isok(const char &c) {
return (c>='a'&&c<='z')||c=='P'||c=='B';
}
inline int getint() {
register int x=0;
for(; !isdig(*is); ++is);
for(; isdig(*is); ++is) {
x=x*10+*is-48;
}
return x;
}
inline void putint(int x) {
if(x==0) {
*is++='0';
}
else {
static int s[10], top;
for(top=0; x; x/=10) s[++top]=x%10;
while(top) {
*is++=s[top--]+'0';
}
}
*is++='\n';
}
struct E {
int next, to, id;
}e[N];
inline void add(int x, int y, int id=-1) {
e[++cnt]=(E){ihead[x], y, id}; ihead[x]=cnt;
}
inline void adds(int x, int s) {
for(; x<=tot; x+=x&-x) {
su[x]+=s;
}
}
inline int sum(int x) {
int y=0;
for(; x; x-=x&-x) {
y+=su[x];
}
return y;
}
inline void bfs() {
static int q[N], fr, ta;
fr=ta=0;
q[ta++]=0;
while(fr!=ta) {
int x=q[fr++];
for(int ch=0; ch<26; ++ch) {
if(c[x][ch]) {
int y=c[x][ch];
q[ta++]=y;
if(x==0) {
add(0, y);
continue;
}
fail[y]=c[fail[x]][ch];
add(fail[y], y);
}
else {
c[x][ch]=c[fail[x]][ch];
}
}
}
}
inline void dfs(int x) {
static int fid=0;
FF[x]=++fid;
for(int i=ihead[x]; i; i=e[i].next) {
dfs(e[i].to);
}
LL[x]=fid;
}
inline void getans(int x) {
adds(FF[x], 1);
for(int i=ihead[x]; i; i=e[i].next) {
int y=e[i].to;
ans[e[i].id]=sum(LL[y])-sum(FF[y]-1);
}
for(int ch=0; ch<26; ++ch) {
if(c[x][ch] && (x==0 || c[x][ch]!=c[fail[x]][ch])) {
getans(c[x][ch]);
}
}
adds(FF[x], -1);
}
void init() {
fread(buf, 1, sizeof buf, stdin);
for(; !isok(*is); ++is);
int now=0, n=0, len=0;
for(; isok(*is); ++is) {
if(*is=='P') {
pos[++n]=now;
}
else if(*is=='B') {
now=last[len--];
}
else {
int ch=*is-'a';
if(!c[now][ch]) {
c[now][ch]=++tot;
}
last[++len]=now;
now=c[now][ch];
}
}
bfs();
dfs(0);
}
int main() {
init();
memset(ihead, 0, sizeof(int)*(tot+1));
cnt=0;
int m=getint();
for(int i=0; i<m; ++i) {
int x, y;
x=getint();
y=getint();
add(pos[y], pos[x], i);
}
++tot;
getans(0);
is=buf;
for(int i=0; i<m; ++i) {
putint(ans[i]);
}
fwrite(buf, 1, sizeof(char)*(is-buf), stdout);
return 0;
}
上一篇:Android 的 so 文件加载机制


下一篇:DW Basic Knowledge1