【BZOJ3881】[Coci2015]Divljak
Description
Alice有n个字符串S_1,S_2...S_n,Bob有一个字符串集合T,一开始集合是空的。
接下来会发生q个操作,操作有两种形式:
“1 P”,Bob往自己的集合里添加了一个字符串P。
“2 x”,Alice询问Bob,集合T中有多少个字符串包含串S_x。(我们称串A包含串B,当且仅当B是A的子串)
Bob遇到了困难,需要你的帮助。
Input
第1行,一个数n;
接下来n行,每行一个字符串表示S_i;
下一行,一个数q;
接下来q行,每行一个操作,格式见题目描述。
Output
对于每一个Alice的询问,帮Bob输出答案。
Sample Input
3
a
bc
abc
5
1 abca
2 1
1 bca
2 2
2 3
a
bc
abc
5
1 abca
2 1
1 bca
2 2
2 3
Sample Output
1
2
1
2
1
HINT
【数据范围】
1 <= n,q <= 100000;
Alice和Bob拥有的字符串长度之和各自都不会超过 2000000;
字符串都由小写英文字母组成。
题解:我们先求出fail树,很容易发现当我们在fail树上遍历T_i字符串时,所经过的节点和它的祖先都会被T_i包含,我们只需要将这些点的权值全部+1,但我们并不能只是将每个点到根的路径上的点权全都+1,因为这样可能导致重复计算
所以,我们要解决的问题是如何将这些点到根节点的路径的并集+1,就是求树链的并,具体方法:
将所有经过的点按照DFS序排序,然后求出相邻两点间的LCA
将所有点到根的路径上的点权+1,再讲所有LCA到根的路径上的点权-1
这个可以用树剖+树状数组搞定
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
int n,m,tot,cnt;
const int maxn=2000010;
struct node
{
int ch[30],fail;
}p[maxn];
int P[maxn],Q[maxn],s[maxn],son[maxn],dep[maxn],to[maxn],head[maxn],next[maxn],top[maxn],siz[maxn];
int v[maxn],vis[maxn],pos[maxn];
queue<int> q;
char str[maxn];
bool cmp(int a,int b)
{
return P[a]<P[b];
}
void add(int a,int b)
{
to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
}
void build()
{
q.push(1);
int i,t,u;
while(!q.empty())
{
u=q.front(),q.pop();
for(i=0;i<26;i++)
{
if(!p[u].ch[i]) continue;
q.push(p[u].ch[i]);
if(u==1)
{
p[p[u].ch[i]].fail=1;
continue;
}
int t=p[u].fail;
while(!p[t].ch[i]&&t) t=p[t].fail;
if(t) p[p[u].ch[i]].fail=p[t].ch[i];
else p[p[u].ch[i]].fail=1;
}
}
}
void dfs1(int x)
{
siz[x]=1;
for(int i=head[x];i!=-1;i=next[i])
{
dep[to[i]]=dep[x]+1;
dfs1(to[i]);
siz[x]+=siz[to[i]];
if(siz[to[i]]>siz[son[x]]) son[x]=to[i];
}
}
void dfs2(int x,int tp)
{
top[x]=tp,P[x]=++P[0];
if(son[x]) dfs2(son[x],tp);
for(int i=head[x];i!=-1;i=next[i])
if(to[i]!=son[x])
dfs2(to[i],to[i]);
Q[x]=P[0];
}
int lca(int x,int y)
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
x=p[top[x]].fail;
}
if(dep[x]<dep[y]) return x;
return y;
}
void updata(int x,int val)
{
for(int i=x;i<=tot;i+=i&-i) s[i]+=val;
}
int query(int x)
{
int i,ret=0;
for(i=x;i;i-=i&-i) ret+=s[i];
return ret;
}
int main()
{
scanf("%d",&n);
int i,j,a,b,c,u;
tot=1;
for(i=1;i<=n;i++)
{
scanf("%s",str);
u=1,a=strlen(str);
for(j=0;j<a;j++)
{
b=str[j]-'a';
if(!p[u].ch[b]) p[u].ch[b]=++tot;
u=p[u].ch[b];
}
pos[i]=u;
}
build();
memset(head,-1,sizeof(head));
for(i=2;i<=tot;i++) add(p[i].fail,i);
dep[1]=1,dfs1(1),dfs2(1,1);
scanf("%d",&m);
for(i=1;i<=m;i++)
{
scanf("%d",&c);
if(c==1)
{
scanf("%s",str);
u=1,a=strlen(str);
vis[1]=i,v[v[0]=1]=1;
for(j=0;j<a;j++)
{
b=str[j]-'a';
while(!p[u].ch[b]&&u!=1) u=p[u].fail;
u=(p[u].ch[b]>0)?p[u].ch[b]:1;
if(vis[u]!=i) vis[u]=i,v[++v[0]]=u;
}
sort(v+1,v+v[0]+1,cmp);
for(j=1;j<=v[0];j++) updata(P[v[j]],1);
for(j=1;j<v[0];j++) updata(P[lca(v[j],v[j+1])],-1);
}
if(c==2)
{
scanf("%d",&a);
printf("%d\n",query(Q[pos[a]])-query(P[pos[a]]-1));
}
}
return 0;
}