bzoj2157旅游

bzoj2157旅游

题意:

给定有权树,支持单边权修改,路径边权取相反数,路径边权求和,路径边权求最大最小值。

题解:

用link-cut tree

link-cut tree与树链剖分有些类似,都是用某种数据结构维护树链。但也有很大差异:树链剖分是依据子树节点数确定轻重边,一经确定,不能更改,所以用相对静态的线段树维护,常数也较小。而link-cut tree是用来求解动态树问题的,它的链随时可以改变,因此也只能用动态的splay来维护,常数较大。以下简称lct

注意:本数据结构中splay是以节点在lct中的深度为关键字确定顺序的。lct中节点的fa有两个意思,当它不为自己所在splay的根结点时,fa表示其在splay中的父节点,当它为splay中的根节点时,fa表示这整条树链在lct中的父亲节点。

lct有三个基础操作:

1、access(u):使节点到根节点连成一条链,并把链上的分支断开。注意access过程就是不停的把节点旋到splay顶,将它与它的fa(即树链在lct中的fa)连接起来,因为是连接成一个splay,所以其fa节点的c[fa][1]也要修改并update
2、link(u,v):将u所在lct连到v所在lct。先access(u),再splay(u),此时u的fa就是u所在lct的fa,将u的fa置为v就行了
3、cut(u):将u从它所在的lct中脱离。先access(u),再splay(u),此时u的左子节点就是lct上它的实际父亲,将它切了就行。因为是涉及splay的切除,所以c[u][0]也要修改并update
可是有了这些怎么得到或更新任意两点间的信息呢?
我们知道,求两点间的信息一般都要求lca,lct怎么会没这个功能呢?求u,v的lca:先access(u),再access(v),后者在执行的时候最后得到那个t(见程序)就是lca,因为之前x已经和根节点相连了。两次access后,再splay(u),就形成一条从u连到lca且以u为根的树链,此时将u的信息与lca的右子节点信息合并即可。
吐槽:我是傻叉,rotate顺序写乱了,导致fa信息出错死循环re了3发,然后又因为两次access后没有splay又wa了2发,大样例调不出,自己的小样例又测不出错误,花了3h……
发现一个技巧,要让gdb在满足条件时停下来,可以if(条件满足)函数();然后在函数里设断点,就能准确中断了。
代码:
  1 #include <cstdio>
  2 #include <cstring>
  3 #include <algorithm>
  4 #define N 20010
  5 #define inc(i,j,k) for(int i=j;i<=k;i++)
  6 #define INF 0x3fffffff
  7 using namespace std;
  8 
  9 //general
 10 int n,m;
 11 void debug(){
 12     n=n;
 13 }
 14 
 15 //edge
 16 struct e{int t,w,n;}; e es[N*2]; int ess,g[N];
 17 void pe(int f,int t,int w){es[++ess]=(e){t,w,g[f]}; g[f]=ess; es[++ess]=(e){f,w,g[t]}; g[t]=ess;}
 18 
 19 //splay
 20 int fa[N]/*点的父亲,包括在splay中的父亲和所在splay的父亲*/,c[N][2]/*splay中的儿子节点*/;
 21 int sm[N],mx[N],mn[N],v[N],/*区间维护信息*/dt[N],dts/*中间过程用*/; bool tg[N]/*标记*/;
 22 inline bool is_root(int x){return fa[x]==0||(c[fa[x]][0]!=x&&c[fa[x]][1]!=x);}//判断是否为所在splay的根节点
 23 void pushdown(int x){//标记下传。注:本程序中的标记表明本节点已更新子孙未更新 
 24     if(x==0)return;
 25     if(tg[x]){
 26         int l=c[x][0],r=c[x][1]; tg[x]^=1;
 27         if(l){tg[l]^=1; v[l]=-v[l]; sm[l]=-sm[l]; int t=mx[l]; mx[l]=-mn[l]; mn[l]=-t;}
 28         if(r){tg[r]^=1; v[r]=-v[r]; sm[r]=-sm[r]; int t=mx[r]; mx[r]=-mn[r]; mn[r]=-t;}
 29     }
 30 }
 31 void update(int x){
 32     if(x==0)return;
 33     int l=c[x][0],r=c[x][1]; sm[x]=v[x]+sm[l]+sm[r]; mx[x]=max(v[x],max(mx[l],mx[r])); mn[x]=min(v[x],min(mn[l],mn[r]));
 34 }
 35 void rotate(int x){//旋转 
 36     if(x==0||is_root(x))return;
 37     int a=fa[x],b=fa[fa[x]]; bool d=c[a][1]==x,e=c[b][1]==a;
 38     if(! is_root(a))c[b][e]=x;//注意在调用is_root前不能动参数的fa[a]和c[fa[a]][0]c[fa[a]][1],不然会导致错误结果 
 39     if(c[x][!d])fa[c[x][!d]]=a;/*注意别漏*/ fa[x]=b; fa[a]=x; c[a][d]=c[x][!d]; c[x][!d]=a;//注意修改的相对顺序  
 40     update(a); update(x); if(! is_root(x))update(b); 
 41 }
 42 void splay(int x){//伸展 
 43     if(x==0)return;
 44     dts=0; int t=x; 
 45     while(! is_root(t))dt[++dts]=t,t=fa[t]; dt[++dts]=t;//将根结点到x的所有标记一次下传 
 46     while(dts)pushdown(dt[dts]),dts--;
 47     while(! is_root(x)){
 48         if(!is_root(fa[x])) (c[fa[x]][1]==x)^(c[fa[fa[x]]][1]==fa[x])?rotate(x):rotate(fa[x]);
 49         rotate(x);
 50     }
 51 }
 52 
 53 //lct
 54 int access(int x){//lct基本操作,使节点到根节点连成一条链,并把链上的分支断开 
 55     if(x==0)return 0;
 56     int t=0;
 57     while(x){
 58         splay(x); c[x][1]=t; if(t)fa[t]=x;
 59         update(x); t=x; x=fa[x];
 60     }
 61     debug();
 62     return t;
 63 }
 64 void link(int x,int y){//lct基本操作,本程序不用 
 65     if(x==0||y==0)return;
 66     access(x); splay(x); fa[x]=y;
 67 }
 68 void cut(int x){//lct基本操作,本程序不用
 69     if(x==0)return;
 70     access(x); splay(x); if(c[x][0])fa[c[x][0]]=0; c[x][0]=0; update(x);
 71 }
 72 
 73 //command
 74 //注意修改时要根据题目要求将修改查询边权变为程序中的修改查询点权 
 75 void change(int x,int val){//更新权值 
 76     splay(x); v[x]=val; update(x);
 77 }
 78 void rever(int x,int y){//反转 
 79     access(x); int a=access(y); /**/splay(x);/*漏!下同*/
 80     if(x==a){
 81         int r=c[x][1]; tg[r]^=1; v[r]=-v[r]; sm[r]=-sm[r]; int t=mx[r]; mx[r]=-mn[r]; mn[r]=-t; update(x);
 82     }else{
 83         tg[x]^=1; v[x]=-v[x]; sm[x]=-sm[x]; int t=mx[x]; mx[x]=-mn[x]; mn[x]=-t;
 84         int r=c[a][1]; tg[r]^=1; v[r]=-v[r]; sm[r]=-sm[r]; t=mx[r]; mx[r]=-mn[r]; mn[r]=-t;
 85         update(a);
 86     }
 87 }
 88 int querysum(int x,int y){//求和 
 89     access(x); int a=access(y); splay(x);
 90     if(x==a)return sm[c[x][1]];else return sm[x]+sm[c[a][1]];
 91 }
 92 int querymax(int x,int y){//求最大值 
 93     access(x); int a=access(y); splay(x);
 94     if(x==a)return mx[c[x][1]];else return max(mx[x],mx[c[a][1]]);
 95 }
 96 int querymin(int x,int y){//求最小值 
 97     access(x); int a=access(y); splay(x);
 98     if(x==a)return mn[c[x][1]];else return min(mn[x],mn[c[a][1]]);
 99 }
100 
101 int num[N];
102 void dfs(int x){//建树,并将边权转为点权 ,本程序用边的终点点权表示这条边的边权 
103     for(int i=g[x];i!=-1;i=es[i].n)if(es[i].t!=fa[x]){
104         fa[es[i].t]=x; v[es[i].t]=sm[es[i].t]=mx[es[i].t]=mn[es[i].t]=es[i].w; dfs(es[i].t);
105     }
106 }
107 int main(){
108     //freopen("big.txt","r",stdin); freopen("big.out","w",stdout);
109     memset(num,0,sizeof(num)); ess=0; memset(g,-1,sizeof(g)); memset(fa,0,sizeof(fa));
110     scanf("%d",&n); inc(i,1,n-1){int a,b,c; scanf("%d%d%d",&a,&b,&c); pe(a+1,b+1,c); num[i]=b+1;}
111     dfs(1); 
112     memset(c,0,sizeof(c)); memset(tg,0,sizeof(tg)); mn[0]=INF; mx[0]=-INF; scanf("%d",&m); char s[10];
113     inc(i,1,m){
114         int a,b; scanf("%s%d%d",s,&a,&b);
115         if(s[0]=='C')change(num[a],b);
116         if(s[0]=='N')a++,b++,rever(a,b);
117         if(s[0]=='S')a++,b++,printf("%d\n",querysum(a,b));
118         if(s[1]=='A')a++,b++,printf("%d\n",querymax(a,b));
119         if(s[1]=='I')a++,b++,printf("%d\n",querymin(a,b));
120     }
121     return 0;
122 }

 

20151215

-----------------------------------------------------------------------------------------------------

题解:

用树链剖分重写一下本题。树链剖分主要分3个步骤:

1、第一次dfs,求出各节点的子树大小、权值、深度等信息。

2、第二次dfs,求出各节点所在树链的头,以及各节点的中儿子、在所有树链依次相连组成的大链中的位置

3、将所有树链依次相连组成的大链建成一棵全局线段树

如何求lca(u,v)?

不停循环这样一个过程:u和v哪个深度大,就让它向上爬一条重链再爬一条轻边,直到两个所在链头节点相等(即在同一条树链)。此时,深度小的那个就是lca。

如何维护(查询)(u,v)之间信息?

求lca(u,v),然后对u不停循环这个过程:维护u到所在树链头的信息,接着u爬一条重链再一条轻边,直到树链头深度小于lca(此时u与lca在同一条树链),然后维护u到lca的信息。对v重复相同操作。

程序中我之所以存了每个节点的重儿子,是因为题意求边权,根据我的程序中边权与点权的转换方式,lca的权值不能被维护(查询),所以“维护u到lca的信息”就变成“维护u到lca重儿子的信息(如果此时u=lca,不能执行此操作)”

吐槽:线段树真耗空间,开的数组大小必须是点数的4倍!因为这个我re了5发。同时线段树的pushdown是边查询(修改)边执行的,需要注意。

对比链剖和link-cut tree,发现lct完胜!?不管在代码长度还是时间空间复杂度上都是lct更优。不是说线段树的常数比splay小得多吗?个人认为链剖输在一下几个方面:

1、链剖编程复杂度高,线段树要写一堆操作,链剖本身还要写一堆操作,同时链剖的操作还要写两个循环,大大增大了代码长度。

2、链剖空间复杂度高,4倍点数。这也是引起速度慢的一个重要原因,开大数组在一定程度上会增大程序的耗时。

总结:本题数据不是很大,链剖的适用范围应该是数据比较大的题,因为此时对程序本身常数的要求要高过空间引起的常数。

根本原因:我太弱了,肯定是我链剖写残才会这么慢!

代码:

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <algorithm>
  4 #define N 20010
  5 #define inc(i,j,k) for(int i=j;i<=k;i++)
  6 #define INF 0x3fffffff
  7 using namespace std;
  8 
  9 //gerneral
 10 int n,m;
 11 
 12 //edge
 13 struct e{int t,w,n;}; e es[N*2]; int ess,g[N];
 14 inline void pe(int f,int t,int w){es[++ess]=(e){t,w,g[f]}; g[f]=ess; es[++ess]=(e){f,w,g[t]}; g[t]=ess;}
 15 
 16 //segment tree
 17 int sm[N*4],mx[N*4],mn[N*4]/*节点信息*/,v[N][2]/*中间数组*/,l[N*4],r[N*4]/*区间*/;
 18 int lc[N*4],rc[N*4]/*左右儿子*/; bool tg[N*4]/*标记*/;
 19 void update(int x){
 20     if(x==0)return;
 21     sm[x]=sm[lc[x]]+sm[rc[x]]; mx[x]=max(mx[lc[x]],mx[rc[x]]); mn[x]=min(mn[lc[x]],mn[rc[x]]);
 22 }
 23 void pushdown(int x){//标记意义同lct 
 24     if(x==0||!tg[x])return; int L=lc[x],R=rc[x]; tg[x]^=1;
 25     if(L){tg[L]^=1; sm[L]=-sm[L]; int t=mx[L]; mx[L]=-mn[L]; mn[L]=-t;}
 26     if(R){tg[R]^=1; sm[R]=-sm[R]; int t=mx[R]; mx[R]=-mn[R]; mn[R]=-t;}
 27 }
 28 void build(int x,int L,int R){//建线段树 
 29     l[x]=L; r[x]=R;
 30     if(L==R)sm[x]=mx[x]=mn[x]=v[L][1],lc[x]=rc[x]=tg[x]=0;else{
 31         int M=(L+R)>>1; lc[x]=x<<1; rc[x]=x<<1|1; build(lc[x],L,M); build(rc[x],M+1,R); tg[x]=0; update(x);
 32     }
 33 }
 34 void change(int x,int nod,int val){//线段树点修改 
 35     pushdown(x);
 36     if(l[x]==r[x])sm[x]=mx[x]=mn[x]=val;else{
 37         int M=(l[x]+r[x])>>1; nod<=M?change(lc[x],nod,val):change(rc[x],nod,val); update(x);
 38     }
 39 }
 40 void rever(int x,int ql,int qr){//线段树区间修改 
 41     pushdown(x);
 42     int M=(l[x]+r[x])>>1; if(ql<=l[x]&&r[x]<=qr){tg[x]^=1; sm[x]=-sm[x]; int t=mx[x]; mx[x]=-mn[x]; mn[x]=-t; return;}
 43     if(ql<=M)rever(lc[x],ql,qr); if(qr>M)rever(rc[x],ql,qr); update(x);
 44 }
 45 int querysum(int x,int ql,int qr){//线段树求和 
 46     pushdown(x);
 47     int ret=0,M=(l[x]+r[x])>>1; if(ql<=l[x]&&r[x]<=qr)return sm[x];
 48     if(ql<=M)ret+=querysum(lc[x],ql,qr); if(qr>M)ret+=querysum(rc[x],ql,qr);
 49     return ret;
 50 }
 51 int querymax(int x,int ql,int qr){//线段树求最大值 
 52     pushdown(x);
 53     int ret=-INF,M=(l[x]+r[x])>>1; if(ql<=l[x]&&r[x]<=qr)return mx[x];
 54     if(ql<=M)ret=max(ret,querymax(lc[x],ql,qr)); if(qr>M)ret=max(ret,querymax(rc[x],ql,qr));
 55     return ret;
 56 }
 57 int querymin(int x,int ql,int qr){//线段树求最小值 
 58     pushdown(x);
 59     int ret=INF,M=(l[x]+r[x])>>1; if(ql<=l[x]&&r[x]<=qr)return mn[x];
 60     if(ql<=M)ret=min(ret,querymin(lc[x],ql,qr)); if(qr>M)ret=min(ret,querymin(rc[x],ql,qr));
 61     return ret;
 62 }
 63 
 64 //tree chain apart
 65 int pos[N]/*点在线段树中位置*/,top[N]/*树链头节点*/,fa[N],sz[N]/*子树大小*/,sgs,dep[N]/*深度*/,ps[N]/*重儿子*/;
 66 void dfs(int x){//得到节点的父亲、子树大小、权值、深度 
 67     sz[x]=1;
 68     for(int i=g[x];i!=0;i=es[i].n)if(es[i].t!=fa[x]){
 69         fa[es[i].t]=x; dep[es[i].t]=dep[x]+1; v[es[i].t][0]=es[i].w; dfs(es[i].t); sz[x]+=sz[es[i].t];
 70     }
 71 }
 72 void buildchain(int x,int tp/*当前节点所在树链头*/){//通过子树大小构造树链 
 73     pos[x]=++sgs; v[sgs][1]=v[x][0]; top[x]=tp; ps[x]=0;
 74     for(int i=g[x];i!=0;i=es[i].n)if(es[i].t!=fa[x]){
 75         if(sz[es[i].t]>sz[ps[x]])ps[x]=es[i].t;
 76     }
 77     if(ps[x]==0)return;
 78     buildchain(ps[x],tp);
 79     for(int i=g[x];i!=0;i=es[i].n)if(es[i].t!=fa[x]){
 80         if(es[i].t!=ps[x])buildchain(es[i].t,es[i].t);
 81     }
 82 }
 83 void init(){
 84     memset(fa,0,sizeof(fa)); dep[1]=1; dfs(1);  
 85     sgs=0; mx[0]=-INF; mn[0]=INF; sz[0]=0; buildchain(1,1); build(1,1,sgs);
 86 }
 87 int lca(int x,int y){
 88     for(;top[x]!=top[y];x=fa[top[x]]){if(dep[top[x]]<dep[top[y]])swap(x,y);}
 89     return dep[x]<dep[y]?x:y;
 90 }
 91 void solvechange(int x,int val){change(1,pos[x],val);}
 92 void solverever(int x,int y){
 93     if(x==y)return; int a=lca(x,y); 
 94     while(dep[top[x]]>dep[a])rever(1,pos[top[x]],pos[x])/*注意参数顺序,下同*/,x=fa[top[x]];
 95     if(a!=x)rever(1,pos[ps[a]],pos[x]);
 96     while(dep[top[y]]>dep[a])rever(1,pos[top[y]],pos[y]),y=fa[top[y]];
 97     if(a!=y)rever(1,pos[ps[a]],pos[y]);
 98 }
 99 int solvesum(int x,int y){
100     if(x==y)return 0; int a=lca(x,y),ans=0; 
101     while(dep[top[x]]>dep[a])ans+=querysum(1,pos[top[x]],pos[x]),x=fa[top[x]];
102     if(a!=x)ans+=querysum(1,pos[ps[a]],pos[x]);
103     while(dep[top[y]]>dep[a])ans+=querysum(1,pos[top[y]],pos[y]),y=fa[top[y]];
104     if(a!=y)ans+=querysum(1,pos[ps[a]],pos[y]);
105     return ans;
106 }
107 int solvemax(int x,int y){
108     if(x==y)return 0; int a=lca(x,y),ans=-INF; 
109     while(dep[top[x]]>dep[a])ans=max(ans,querymax(1,pos[top[x]],pos[x])),x=fa[top[x]];
110     if(a!=x)ans=max(ans,querymax(1,pos[ps[a]],pos[x]));
111     while(dep[top[y]]>dep[a])ans=max(ans,querymax(1,pos[top[y]],pos[y])),y=fa[top[y]];
112     if(a!=y)ans=max(ans,querymax(1,pos[ps[a]],pos[y]));
113     return ans;
114 }
115 int solvemin(int x,int y){
116     if(x==y)return 0; int a=lca(x,y),ans=INF; 
117     while(dep[top[x]]>dep[a])ans=min(ans,querymin(1,pos[top[x]],pos[x])),x=fa[top[x]];
118     if(a!=x)ans=min(ans,querymin(1,pos[ps[a]],pos[x]));
119     while(dep[top[y]]>dep[a])ans=min(ans,querymin(1,pos[top[y]],pos[y])),y=fa[top[y]];
120     if(a!=y)ans=min(ans,querymin(1,pos[ps[a]],pos[y]));
121     return ans;
122 }
123 
124 //main
125 int num[N];
126 int main(){
127     //freopen("zs.txt","r",stdin); freopen("zs.out","w",stdout);
128     scanf("%d",&n); memset(g,0,sizeof(g)); ess=0;
129     inc(i,1,n-1){int a,b,c; scanf("%d%d%d",&a,&b,&c); a++; b++; pe(a,b,c); num[i]=b;}
130     init(); scanf("%d",&m); char s[10];
131     inc(i,1,m){
132         int a,b; scanf("%s%d%d",s,&a,&b);
133         if(s[0]=='C')solvechange(num[a],b);
134         if(s[0]=='N')a++,b++,solverever(a,b);
135         if(s[0]=='S')a++,b++,printf("%d\n",solvesum(a,b));
136         if(s[1]=='A')a++,b++,printf("%d\n",solvemax(a,b));
137         if(s[1]=='I')a++,b++,printf("%d\n",solvemin(a,b));
138     }
139     return 0;
140 }

 

20151218

 
posted @ 2015-12-15 22:01  YuanZiming  阅读(282)  评论(0编辑  收藏  举报