bzoj 4545: DQS的trie

时间:2024-04-05 20:34:29

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中的出现次数。

Solution

比较直观

操作 \(1\) ,就是维护 \(\sum len[x]-len[fa[x]]\) ,改变父亲关系时加减一下就好了

然而在加入一个 \(nt\) 的时候,把式子列出来发现直接抵消了,\(QAQ\)

实际上维护一下 \(cur\) 的变化就好了

操作 \(2,3\) 和 \(substring\) 一题类似, \(LCT\) 维护链加法就好了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
template<class T>void gi(T &x){
int f;char c;
for(f=1,c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')f=-1;
for(x=0;c<='9'&&c>='0';c=getchar())x=x*10+(c&15);x*=f;
}
const int N=2e5+10;
namespace lct{
int fa[N],ch[N][2],w[N],la[N];
inline bool isrt(int x){return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;}
inline void mark(int x,int t){la[x]+=t;w[x]+=t;}
inline void pushdown(int x){
if(!la[x])return ;
mark(ch[x][0],la[x]);mark(ch[x][1],la[x]);la[x]=0;
}
inline void rotate(int x){
int y=fa[x];bool t=ch[y][1]==x;
ch[y][t]=ch[x][!t];fa[ch[y][t]]=y;
ch[x][!t]=y;fa[x]=fa[y];
if(!isrt(y))ch[fa[y]][ch[fa[y]][1]==y]=x;
fa[y]=x;
}
inline void Push(int x){
if(!isrt(x))Push(fa[x]);
pushdown(x);
}
inline void splay(int x){
Push(x);
while(!isrt(x)){
int y=fa[x],p=fa[y];
if(isrt(y))rotate(x);
else if((ch[p][0]==y)==(ch[y][0]==x))rotate(y),rotate(x);
else rotate(x),rotate(x);
}
}
inline void access(int x){
int y=0;
while(x)splay(x),ch[x][1]=y,x=fa[y=x];
}
inline void link(int x,int y){
fa[x]=y;access(y);splay(y);splay(x);mark(y,w[x]);
}
inline void cut(int x){
access(x);splay(x);mark(ch[x][0],-w[x]);ch[x][0]=fa[ch[x][0]]=0;
}
inline int qry(int x){splay(x);return w[x];}
}
int n,Q,head[N],nxt[N*2],to[N*2],num=0,c[N],las[N],cnt=1,cur;
ll ans=0;int ch[N][26],fa[N],len[N];char s[N];
inline void link(int x,int y,int z){
nxt[++num]=head[x];to[num]=y;head[x]=num;c[num]=z;
nxt[++num]=head[y];to[num]=x;head[y]=num;c[num]=z;
}
inline void ins(int c,int p){
cur=++cnt;len[cur]=len[p]+1;lct::w[cur]=1;
for(;p && !ch[p][c];p=fa[p])ch[p][c]=cur;
if(!p)fa[cur]=1,lct::link(cur,1);
else{
int q=ch[p][c];
if(len[p]+1==len[q])fa[cur]=q,lct::link(cur,q);
else{
int nt=++cnt;len[nt]=len[p]+1;
memcpy(ch[nt],ch[q],sizeof(ch[q]));
lct::cut(q);lct::link(nt,fa[q]);lct::link(q,nt);lct::link(cur,nt);
fa[nt]=fa[q];fa[q]=fa[cur]=nt;
for(;p && ch[p][c]==q;p=fa[p])ch[p][c]=nt;
}
}
ans+=len[cur]-len[fa[cur]];
}
inline void bfs(int S){
queue<int>Q;Q.push(S);
while(!Q.empty()){
int x=Q.front();Q.pop();
for(int i=head[x],u;i;i=nxt[i]){
if(las[u=to[i]])continue;
ins(c[i],las[x]);las[u]=cur;Q.push(u);
}
}
}
inline int solve(){
scanf("%s",s+1);
int le=strlen(s+1),p=1;
for(int i=1;i<=le;i++){
int c=s[i]-'a';
if(ch[p][c])p=ch[p][c];
else return 0;
}
return lct::qry(p);
}
int main(){
freopen("pp.in","r",stdin);
freopen("pp.out","w",stdout);
gi(n);gi(n);
int x,y,op,p,q;char ch[2];
for(int i=1;i<n;i++)gi(x),gi(y),scanf("%s",ch),link(x,y,ch[0]-'a');
bfs(las[1]=1);gi(Q);
while(Q--){
gi(op);
if(op==1)printf("%lld\n",ans);
else if(op==2){
gi(p);gi(q);
for(int i=1;i<q;i++){
gi(x),gi(y),scanf("%s",ch);
link(x,y,ch[0]-'a');
}
bfs(p);
}else printf("%d\n",solve());
}
return 0;
}