题意:
给定有权树,支持单边权修改,路径边权取相反数,路径边权求和,路径边权求最大最小值。
题解:
用link-cut tree
link-cut tree与树链剖分有些类似,都是用某种数据结构维护树链。但也有很大差异:树链剖分是依据子树节点数确定轻重边,一经确定,不能更改,所以用相对静态的线段树维护,常数也较小。而link-cut tree是用来求解动态树问题的,它的链随时可以改变,因此也只能用动态的splay来维护,常数较大。以下简称lct
注意:本数据结构中splay是以节点在lct中的深度为关键字确定顺序的。lct中节点的fa有两个意思,当它不为自己所在splay的根结点时,fa表示其在splay中的父节点,当它为splay中的根节点时,fa表示这整条树链在lct中的父亲节点。
lct有三个基础操作:
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