4545: DQS的trie
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 206 Solved: 76
[Submit][Status][Discuss]
Description
DQS的自家阳台上种着一棵颗粒饱满、颜色纯正的trie。
DQS的trie非常的奇特,它初始有n0个节点,n0-1条边,每条边上有一个字符。并且,它拥有极强的生长力:某个i时刻,某个节点就会新生长出一颗子树,它拥有si个节点且节点之间的边上有一个字符,并且新生长出来的子树也是一个树结构。然而因为是新长出来的,根据生活常识可知si必定不会大于i时刻之前的树的大小。
DQS定义trie的子串为从根节点(1号节点)往下走到所有节点所构成的字符串的所有的后缀。DQS身为一个单身doge,常常取出其中一个子串送给妹子,然而他并不希望送给妹子两个相同的子串,所以他非常关心当前trie的本质不同的子串数目。
DQS有时还会去商店购买子串,若他在商店看上某个子串,他希望得知这个子串是否在自家阳台的trie上已经出现,若出现则出现了多少次。如果出现了,他就可以直接回家取trie上的子串辣!
然而DQS身为一个蒟蒻,看着自家阳台的trie树一天天在长大,他被如此众多的节点弄得眼花缭乱,于是他找到了IOI2016Au的你。他会告诉你自家trie树的成长历程,他希望你能够对于每一次询问都做出正确回复。
Input
第一行输入一个整数id,代表测试点编号。
接下来一行输入一个整数n0,表示初始树的大小。
接下来n0-1行,每行两个整数u,v和一个字符c,表示u号节点和v号节点之间有一条边,边上的字母为c。
接下来输入m表示有m组操作。
对于每一组,第一行输入一个整数opt。
若opt=1,则是一组询问,询问当前trie的本质不同的子串数目是多少。
若opt=2,则后面跟两个整数rt,si,表示以点rt为根向下长出一个子树,大小为si。
接下来si-1行,每行两个整数u,v和一个字符c,表示u号节点和v号节点之间有一条边,边上的字母为c。若长出子树之前当前树的大小是n,则这si-1点的编号分别为n+1,n+2…n+si-1。
若opt=3,则是一组询问,后面输入一个字符串S,询问字符串S在当前trie中的出现次数。
Output
对于每个opt=1或3,输出一行表示答案。
Sample Input
1
4
1 2 a
1 3 b
2 4 b
6
1
2 2 4
2 5 b
2 6 c
5 7 b
1
3 ab
2 6 3
6 8 a
6 9 b
1
4
1 2 a
1 3 b
2 4 b
6
1
2 2 4
2 5 b
2 6 c
5 7 b
1
3 ab
2 6 3
6 8 a
6 9 b
1
Sample Output
3
7
2
11
【数据范围及提示】
第一个询问,本质不同的子串是 a,b,ab。
第二个询问,本质不同的子串是 a,b,c,ab,ac,bb,abb。
第三个询问,ab出现次数是 2。
第四个询问,本质不同的子串是 a,b,c,ab,ac,ca,cb,bb,abb,aca,acb。
opt=1或3时对原树不做修改,只是询问。
每次opt=2,会增加si-1个节点,因为有一个节点是原树上作为新树的根出现的。
数据中,对于链的部分分,满足端点为根节点,每次新建子树都从尾部插入。
对于全部数据,保证从始至终每条边上的字符均为小写字母’a’或’b’或’c’。
n是最终树的大小,N<=100000,M<=100000,Si<=当前树的大小
7
2
11
【数据范围及提示】
第一个询问,本质不同的子串是 a,b,ab。
第二个询问,本质不同的子串是 a,b,c,ab,ac,bb,abb。
第三个询问,ab出现次数是 2。
第四个询问,本质不同的子串是 a,b,c,ab,ac,ca,cb,bb,abb,aca,acb。
opt=1或3时对原树不做修改,只是询问。
每次opt=2,会增加si-1个节点,因为有一个节点是原树上作为新树的根出现的。
数据中,对于链的部分分,满足端点为根节点,每次新建子树都从尾部插入。
对于全部数据,保证从始至终每条边上的字符均为小写字母’a’或’b’或’c’。
n是最终树的大小,N<=100000,M<=100000,Si<=当前树的大小
好久没打数据结构了
感觉代码力逐渐归零...
本质不同串数 sigma(mx[i]-mx[par[i]])
子串出现次数|Right(ST(S))|
用LCT动态维护后缀自动机
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
return x*f;
}
inline int read_ch()
{char ch=getchar();while(ch<'a'||ch>'c')ch=getchar();return ch-'a';}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}
const int N=200100;
ll tot;
struct LCT
{
int ch[N<<1][2],fa[N<<1];
ll w[N],tag[N];
ll& operator [](int x){return w[x];}
inline bool isroot(int x)
{return ch[fa[x]][0]!=x && ch[fa[x]][1]!=x;}
inline void pushdown(int x)
{
if(tag[x])
{
int ls=ch[x][0],rs=ch[x][1];
if(ls) w[ls]+=tag[x],tag[ls]+=tag[x];
if(rs) w[rs]+=tag[x],tag[rs]+=tag[x];
tag[x]=0;
}
}
void getdown(int x)
{if(!isroot(x)) getdown(fa[x]);pushdown(x);}
void rotate(int x)
{
int y=fa[x],z=fa[y],l,r;
l=(ch[y][1]==x);r=l^1;
if(!isroot(y)) ch[z][ch[z][1]==y]=x;
fa[y]=x;fa[x]=z;fa[ch[x][r]]=y;
ch[y][l]=ch[x][r];ch[x][r]=y;
}
void splay(int x)
{
getdown(x);
int y,z;
while(!isroot(x))
{
y=fa[x],z=fa[y];
if(!isroot(y))
{
if((ch[y][0]==x)^(ch[z][0]==y)) rotate(x);
else rotate(y);
}
rotate(x);
}
}
void access(int x)
{int t(0);while(x){splay(x);ch[x][1]=t;t=x;x=fa[x];}}
void rever(int x)
{access(x);splay(x);}
void link(int x,int y)
{fa[x]=y;rever(y);w[y]+=w[x];tag[y]+=w[x];}
void cut(int x)
{rever(x);int y=ch[x][0];ch[x][0]=0;fa[y]=0;if(y) w[y]-=w[x],tag[y]-=w[x];}
ll query(int x)
{splay(x);return w[x];}
};
struct SAM
{
LCT lct;
int sz,root;
int trans[N<<1][3],par[N<<1],mx[N<<1];
SAM(){sz=root=1;}
int insert(int p,int x)
{
int np=++sz;
lct[np]=1;
mx[np]=mx[p]+1;
while(p && !trans[p][x])
trans[p][x]=np,p=par[p];
if(!p) par[np]=root,tot+=mx[np],lct.link(np,root);
else
{
int q=trans[p][x];
if(mx[q]==mx[p]+1) par[np]=q,tot+=mx[np]-mx[q],lct.link(np,q);
else
{
int nq=++sz;
mx[nq]=mx[p]+1;
tot-=mx[q]-mx[par[q]];
memcpy(trans[nq],trans[q],sizeof(trans[q]));
lct.link(nq,par[q]);lct.cut(q);
par[nq]=par[q];
lct.link(q,nq);lct.link(np,nq);
par[q]=par[np]=nq;
tot+=mx[q]-mx[par[q]]+mx[nq]-mx[par[nq]]+mx[np]-mx[par[np]];
while(p && trans[p][x]==q)
trans[p][x]=nq,p=par[p];
}
}
return np;
}
void solve(char *s)
{
int now(1);
for(int i=1;s[i];++i) now=trans[now][s[i]-'a'];
if(!now){puts("0");return ;}
print(lct.query(now));puts("");
}
}sam;
int last[N],ecnt;
struct EDGE{int to,nt,val;}e[N<<1];
inline void add(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
int fa[N],pos[N];
void dfs(int u)
{
for(int i=last[u],v;i;i=e[i].nt)
if(fa[u]!=(v=e[i].to))
fa[v]=u,pos[v]=sam.insert(pos[u],e[i].val),dfs(v);
last[u]=0;
}
char s[N];
int main()
{freopen("trie5.in","r",stdin);
register int rt,si,i,u,v,val,Q,opt;
si=read();si=read();
for(i=1;i<si;++i)
u=read(),v=read(),val=read_ch(),
add(u,v,val),add(v,u,val);
pos[1]=1;dfs(1);ecnt=0;
Q=read();
while(Q--)
{
opt=read();
switch(opt)
{
case 1: print(tot);puts("");break;
case 2:
rt=read();si=read();
for(i=1;i<si;++i)
u=read(),v=read(),val=read_ch(),
add(u,v,val),add(v,u,val);
dfs(rt);ecnt=0;
break;
case 3: scanf("%s",s+1);sam.solve(s);break;
}
}
return 0;
}