COJ 1006 树上操作

传送门：http://oj.cnuschool.org.cn/oj/home/problem.htm?problemID=979

WZJ的数据结构（六）

难度级别：D； 运行时间限制：1000ms； 运行空间限制：51200KB； 代码长度限制：2000000B

试题描述

给你一棵N个节点的无根树，边之间有权值。请你设计一个数据结构，进行以下两种操作：

1.修改：给你a、b，将第a条边的权值改为b。

2.询问：给你a、b，输出从a到b路径上的最大值、最小值与权值和，如果a=b，输出"error"。

输入

第一行为一个正整数N。
接下来N-1行为每一条边，每行3个正整数a，b，c，表示有一条从a到b权值为c的边（从1开始编号）。
第N+1行为一个正整数Q,表示Q次操作。
接下来Q行为每一次询问，每行3个正整数t，a，b，若t=1表示询问，t=0表示修改。

输出

对于每一次询问，输出三个正整数，即最大值、最小值与权值和，如果a=b,输出"error"。

输入示例

5 
1 2 4
1 4 5
3 4 6
2 5 7
4
1 1 5
1 2 5
0 1 2
1 1 5

输出示例

7 4 11
7 7 7
7 2 9

其他说明

1<=N,Q<=50000
1<=c<=1000
询问中的1<=a,b<=N
修改中的1<=a<=N-1，1<=b<=1000

题解：懒得写树链剖分了LCT走起

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<algorithm>
 #include<queue>
 #include<cstring>
 #define PAU putchar(' ')
 #define ENT putchar('\n')
 using namespace std;
 const int maxn=+,inf=-1u>>;
 inline int read(){
     int x=,sig=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')sig=-;ch=getchar();}
     while(isdigit(ch))x=*x+ch-'',ch=getchar();
     return x*=sig;
 }
 inline void write(int x){
     if(x==){putchar('');return;}if(x<)putchar('-'),x=-x;
     int len=,buf[];while(x)buf[len++]=x%,x/=;
     for(int i=len-;i>=;i--)putchar(buf[i]+'');return;
 }
 struct node {
     int x,mi,mx,sm,set;bool rev;
     node *ch[],*fa;
     inline void add_rev_tag(){
         swap(ch[],ch[]);rev^=;return;
     }
     inline void add_set_tag(int a){
         x=set=a;
         if(mi!=inf) mi=set;
         if(mx!=-inf) mx=set;
     }
     inline void down(){
         if(rev){
             if(ch[]) ch[]->add_rev_tag();
             if(ch[]) ch[]->add_rev_tag();
             rev=;
         }
         if(set){
             if(ch[]) ch[]->add_set_tag(set);
             if(ch[]) ch[]->add_set_tag(set);
             set=;
         }
         return;
     }
     inline void update(){
         sm=x;mi=inf;mx=-inf;
         if(ch[]) sm+=ch[]->sm,mi=min(mi,ch[]->mi),mx=max(mx,ch[]->mx);
         if(ch[]) sm+=ch[]->sm,mi=min(mi,ch[]->mi),mx=max(mx,ch[]->mx);
         if(!x) return;
         mi=min(x,mi);
         mx=max(x,mx);
         return;
     }
 }lct[maxn];
 inline int get_parent(node *x,node *&fa){return (fa=x->fa)?fa->ch[]==x?:fa->ch[]==x?:-:-;}
 inline void rotate(node *x){
     int t1,t2;
     node *fa,*gfa;
     t1=get_parent(x,fa);
     t2=get_parent(fa,gfa);
     if ((fa->ch[t1]=x->ch[t1^])) fa->ch[t1]->fa=fa;
     x->ch[t1^]=fa;fa->fa=x;x->fa=gfa;
     if (t2!=-) gfa->ch[t2]=x;
     fa->update();return;
 }
 inline void pushdown(node *x){
     static node *stack[maxn];
     int cnt=;
     while(){
         stack[cnt++]=x;
         node *fa=x->fa;
         if (!fa || (fa->ch[]!=x && fa->ch[]!=x)) break;
         x=fa;
     }
     while(cnt--) stack[cnt]->down();
     return;
 }
 inline node * splay(node *x){
     pushdown(x);
     while(){
         int t1,t2;
         node *fa,*gfa;
         t1=get_parent(x,fa);
         if(t1==-) break;
         t2=get_parent(fa,gfa);
         if(t2==-){
             rotate(x);break;
         } else if (t1==t2){
             rotate(fa);rotate(x);
         } else{
             rotate(x);rotate(x);
         }
     }
     x->update();
     return x;
 }
 inline node * access(node *x){
     node *ret=NULL;
     while (x) splay(x)->ch[]=ret,(ret=x)->update(),x=x->fa;
     return ret;
 }
 inline void makeroot(int x){access(lct+x)->add_rev_tag();}
 inline void link(int u,int v){
     makeroot(u);splay(lct+u)->fa=lct+v;return;
 }
 int n,q;
 int main(){
     n=read();
     int i;
     int lim=n<<;
     for(i=;i<=lim;i++) {
         lct[i]=(node){,inf,-inf,,,};
     }
     for(i=;i<n;i++){
         int u,v,w;
         u=read();v=read();
         link(u,i+n);link(v,i+n);lct[i+n].add_set_tag(read());
     }
     q=read();
     int x,y,c;
     while(q--){
         c=read();x=read();y=read();
         if(c==) makeroot(x+n),access(x+n+lct)->add_set_tag(y);
         else{
             if(x==y){puts("error");continue;}
             makeroot(x);node*t=access(y+lct);
             write(t->mx);PAU;write(t->mi);PAU;write(t->sm);ENT;
         }
     }
     return ;
 }

搜索

复制