题意
DQS的自家阳台上种着一棵颗粒饱满、颜色纯正的trie。
DQS的trie非常的奇特,它初始有n0个节点,n0-1条边,每条边上有一个字符。并且,它拥有极强的生长力:某个i时刻,某个节点就会新生长出一颗子树,它拥有si个节点且节点之间的边上有一个字符,并且新生长出来的子树也是一个树结构。然而因为是新长出来的,根据生活常识可知si必定不会大于i时刻之前的树的大小。
DQS定义trie的子串为从根节点(1号节点)往下走到所有节点所构成的字符串的所有的后缀。DQS身为一个单身doge,常常取出其中一个子串送给妹子,然而他并不希望送给妹子两个相同的子串,所以他非常关心当前trie的本质不同的子串数目。
DQS有时还会去商店购买子串,若他在商店看上某个子串,他希望得知这个子串是否在自家阳台的trie上已经出现,若出现则出现了多少次。如果出现了,他就可以直接回家取trie上的子串辣!
然而DQS身为一个蒟蒻,看着自家阳台的trie树一天天在长大,他被如此众多的节点弄得眼花缭乱,于是他找到了IOI2016Au的你。他会告诉你自家trie树的成长历程,他希望你能够对于每一次询问都做出正确回复。
n是最终树的大小,N<=100000,M<=100000,Si<=当前树的大小
分析
看完题就知道是要在广义sam上搞事。不难想到用lct来维护parents树,但这里有一种我觉得更加优秀的离线做法。
先把整棵trie搞出来然后建sam。
对于询问1,当新加入一个点,那么它到parents树根上的所有没有被算过的点的贡献都要被计算。由于这一定是以1为根的一棵连通子树,所以我们可以暴力往上跳。
对于询问2,实际就是查询该串对应节点的right集大小。用树状数组来维护一下子树中的right集和即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
const int N=200005;
int n,m,cnt,last[N],id[N],ch[N][3],mx[N],fa[N],c[N],dfn[N],dfn_mx[N],sz,tim,s[N];
LL ans;
struct edge{int to,next,c;}e[N*2];
struct query{int op,s;vector<int> vec;}q[N/2];
char str[N/2];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void addedge(int u,int v,int c)
{
e[++cnt].to=v;e[cnt].c=c;e[cnt].next=last[u];last[u]=cnt;
e[++cnt].to=u;e[cnt].c=c;e[cnt].next=last[v];last[v]=cnt;
}
void bit_ins(int x,int y)
{
while (x<=sz) c[x]+=y,x+=x&(-x);
}
int bit_query(int x)
{
int ans=0;
while (x) ans+=c[x],x-=x&(-x);
return ans;
}
int ins(int ls,int x)
{
if (ch[ls][x])
{
int p=ls,q=ch[p][x];
if (mx[q]==mx[p]+1) return q;
int nq=++sz;mx[nq]=mx[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q];fa[q]=nq;
for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
return nq;
}
int p=ls,np=++sz;mx[np]=mx[p]+1;
for (;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
if (!p) fa[np]=1;
else
{
int q=ch[p][x];
if (mx[q]==mx[p]+1) fa[np]=q;
else
{
int nq=++sz;mx[nq]=mx[p]+1;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q];
fa[q]=fa[np]=nq;
for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
}
}
return np;
}
void fir_dfs(int x,int fa)
{
for (int i=last[x];i;i=e[i].next)
{
int to=e[i].to;
if (to==fa) continue;
id[to]=ins(id[x],e[i].c);
fir_dfs(to,x);
}
}
void sec_dfs(int x)
{
dfn[x]=++tim;
for (int i=last[x];i;i=e[i].next)
if (e[i].to!=fa[x]) sec_dfs(e[i].to);
dfn_mx[x]=tim;
}
void prework()
{
n=read();n=read();
for (int i=1;i<n;i++)
{
int x=read(),y=read();
char ch[2];scanf("%s",ch);
addedge(x,y,ch[0]-'a');
}
m=read();
for (int i=1;i<=m;i++)
{
q[i].op=read();
if (q[i].op==2)
{
int r=read(),s=read()-1;q[i].s=s;
for (int j=1;j<=s;j++)
{
int x=read(),y=read();
char ch[2];scanf("%s",ch);
addedge(x,y,ch[0]-'a');
}
}
else if (q[i].op==3)
{
scanf("%s",str);int len=strlen(str);
for (int j=0;j<len;j++) q[i].vec.push_back(str[j]-'a');
}
}
sz=id[1]=1;
fir_dfs(1,0);
cnt=0;
for (int i=1;i<=sz;i++) last[i]=0,s[i]=mx[i]-mx[fa[i]];
for (int i=2;i<=sz;i++) addedge(fa[i],i,0);
sec_dfs(1);
for (int i=1;i<=n;i++)
{
bit_ins(dfn[id[i]],1);
int x=id[i];
while (s[x]) ans+=s[x],s[x]=0,x=fa[x];
}
}
void solve()
{
for (int i=1;i<=m;i++)
{
if (q[i].op==1) printf("%lld\n",ans);
else if (q[i].op==2)
{
for (int j=n+1;j<=n+q[i].s;j++)
{
bit_ins(dfn[id[j]],1);
int x=id[j];
while (s[x]) ans+=s[x],s[x]=0,x=fa[x];
}
n+=q[i].s;
}
else
{
int now=1;
for (int j=0;j<q[i].vec.size();j++) now=ch[now][q[i].vec[j]];
if (!now) puts("0");
else printf("%d\n",bit_query(dfn_mx[now])-bit_query(dfn[now]-1));
}
}
}
int main()
{
prework();
solve();
return 0;
}