博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
bzoj 4545: DQS的trie
阅读量:5138 次
发布时间:2019-06-13

本文共 3958 字,大约阅读时间需要 13 分钟。

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
using namespace std;typedef long long ll;template
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
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

转载于:https://www.cnblogs.com/Yuzao/p/8901328.html

你可能感兴趣的文章
Spring JDBCTemplate
查看>>
Radon变换——MATLAB
查看>>
Iroha and a Grid AtCoder - 1974(思维水题)
查看>>
gzip
查看>>
转负二进制(个人模版)
查看>>
LintCode-Backpack
查看>>
查询数据库锁
查看>>
[LeetCode] Palindrome Number
查看>>
我对于脚本程序的理解——百度轻应用有感
查看>>
SQL更新某列包含XX的所有值
查看>>
网易味央第二座猪场落户江西 面积超过3300亩
查看>>
面试时被问到的问题
查看>>
spring 事务管理
查看>>
VS2008 去掉msvcr90的依赖
查看>>
当前记录已被另一个用户锁定
查看>>
Node.js 连接 MySQL
查看>>
那些年,那些书
查看>>
注解小结
查看>>
java代码编译与C/C++代码编译的区别
查看>>
Bitmap 算法
查看>>