在对树中数据进行改动的时候需要很多pushdown(具体操作见代码),不然会wa,大概原因和线段树区间修改需要很多pushup是一样的。
这个轻重链的方法特别好用,虽然第一次写树链剖分但是容易理解又有优秀复杂度的结构让人情不自禁orz。
(后来发现很久以前学lca的时候就学了树链剖分只不过忘了,mdzz)
因为忘了去掉测试代码的freopen,re了4次(虽然有三次就算不re也wa),发现一行一行检查对于数据结构的调试很有用(虽然一直找不出来re原因但是我起码避免了很多wa呀)。
代码
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 const int maxn=20100; 8 const long long modn=1000000007; 9 const int minf=1<<30; 10 int n,m;char ch[5]={}; 11 struct nod{ 12 int y,next,v; 13 }e[maxn*2]; 14 int head[maxn]={},tot=0; 15 struct seg{ 16 int mi,mx,sum,c,l,r; 17 int f,rev; 18 seg(){mi=mx=sum=c=l=r=0;f=rev=0;} 19 }t[maxn*4]; 20 int dep[maxn]={},crs[maxn]={},pos[maxn]={},road[maxn]={}; 21 int kid[maxn]={},fa[maxn]={},top[maxn]={}; 22 void init(int x,int y,int v){ 23 e[++tot].v=v;e[tot].y=y; 24 e[tot].next=head[x];head[x]=tot; 25 } 26 int dfs1(int x,int pa){ 27 int y,siz=0,tsn=1,ma=0; 28 fa[x]=pa;dep[x]=dep[pa]+1; 29 for(int i=head[x];i;i=e[i].next){ 30 y=e[i].y; 31 if(y==pa)continue; 32 siz=dfs1(y,x);tsn+=siz; 33 if(siz>ma)ma=siz,kid[x]=y; 34 crs[(i+1)/2]=y; 35 } 36 return tsn; 37 } 38 void dfs2(int x,int pa){ 39 int y,v; 40 pos[x]=++tot;top[x]=pa; 41 if(kid[x])dfs2(kid[x],pa); 42 for(int i=head[x];i;i=e[i].next){ 43 y=e[i].y;v=e[i].v; 44 if(y==kid[x])road[pos[kid[x]]]=v; 45 if(y==kid[x]||y==fa[x])continue; 46 road[tot+1]=v; 47 dfs2(y,y); 48 } 49 } 50 void pushup(int x){ 51 int l=x*2,r=x*2+1; 52 t[x].mx=max(t[l].mx,t[r].mx); 53 t[x].mi=min(t[l].mi,t[r].mi); 54 t[x].sum=t[l].sum+t[r].sum; 55 } 56 void pushdown(int x){ 57 if(t[x].f){ 58 int l=x*2,r=x*2+1; 59 t[x].rev=t[x].f=0; 60 t[x].mx=t[x].mi=t[x].c; 61 t[x].sum=t[x].c*(t[x].r-t[x].l+1); 62 if(t[x].l<t[x].r){ 63 t[l].c=t[r].c=t[x].c; 64 t[l].f=t[r].f=1; 65 t[l].rev=t[r].rev=0; 66 } 67 return; 68 } 69 if(t[x].rev){ 70 int l=x*2,r=x*2+1; 71 t[x].rev=0; 72 swap(t[x].mi,t[x].mx); 73 t[x].mx*=-1;t[x].mi*=-1;t[x].sum*=-1; 74 if(t[x].l<t[x].r){ 75 t[l].c*=-1;t[l].rev^=1; 76 t[r].c*=-1;t[r].rev^=1; 77 } 78 } 79 } 80 void build(int x,int l,int r){ 81 t[x].l=l,t[x].r=r; 82 if(l==r){ 83 t[x].mx=t[x].mi=t[x].sum=road[l]; 84 return; 85 } 86 int mid=(l+r)/2; 87 build(x*2,l,mid); 88 build(x*2+1,mid+1,r); 89 pushup(x); 90 } 91 void cov(int x,int l,int r,int w){ 92 if(t[x].l>=l&&t[x].r<=r){ 93 t[x].f=1;t[x].c=w; pushdown(x); 94 return; 95 } 96 pushdown(x); 97 int mid=(t[x].l+t[x].r)/2; 98 if(l<=mid)cov(x*2,l,r,w); 99 pushdown(x*2); 100 if(r>mid) cov(x*2+1,l,r,w); 101 pushdown(x*2+1); 102 pushup(x); 103 } 104 void revs(int x,int l,int r){ 105 if(t[x].l>=l&&t[x].r<=r){ 106 t[x].c*=-1;t[x].rev^=1; 107 pushdown(x); 108 return; 109 }pushdown(x); 110 int mid=(t[x].l+t[x].r)/2; 111 if(l<=mid) revs(x*2,l,r); 112 pushdown(x*2); 113 if(r>mid) revs(x*2+1,l,r); 114 pushdown(x*2+1); 115 pushup(x); 116 } 117 int mi(int x,int l,int r){ 118 pushdown(x); 119 if(t[x].l>=l&&t[x].r<=r){ 120 return t[x].mi; 121 } 122 int mid=(t[x].l+t[x].r)/2,minn=minf; 123 if(l<=mid) minn=min(minn,mi(x*2,l,r)); 124 if(r>mid) minn=min(minn,mi(x*2+1,l,r)); 125 return minn; 126 } 127 int mx(int x,int l,int r){ 128 pushdown(x); 129 if(t[x].l>=l&&t[x].r<=r){ 130 return t[x].mx; 131 } 132 int mid=(t[x].l+t[x].r)/2,mxn=-minf; 133 if(l<=mid) mxn=max(mxn,mx(x*2,l,r)); 134 if(r>mid) mxn=max(mxn,mx(x*2+1,l,r)); 135 return mxn; 136 } 137 int su(int x,int l,int r){ 138 pushdown(x); 139 if(t[x].l>=l&&t[x].r<=r){ 140 return t[x].sum; 141 } 142 int mid=(t[x].l+t[x].r)/2,sumn=0; 143 if(l<=mid) sumn+=su(x*2,l,r); 144 if(r>mid) sumn+=su(x*2+1,l,r); 145 return sumn; 146 } 147 void mrev(int x,int y){ 148 int a,b; 149 for(a=top[x],b=top[y];a!=b;){ 150 if(dep[a]<dep[b])swap(a,b),swap(x,y); 151 revs(1,pos[a],pos[x]); 152 x=fa[a];a=top[x]; 153 } 154 if(dep[x]>dep[y])swap(x,y); 155 if(x!=y) revs(1,pos[x]+1,pos[y]); 156 } 157 void doit(int x,int y,int k){ 158 int a,b,mxn=-minf,minn=minf,sumn=0; 159 for(a=top[x],b=top[y];a!=b;){ 160 if(dep[a]<dep[b])swap(a,b),swap(x,y); 161 if(k==0) sumn+=su(1,pos[a],pos[x]); 162 else if(k==1) mxn=max(mx(1,pos[a],pos[x]),mxn); 163 else minn=min(mi(1,pos[a],pos[x]),minn); 164 x=fa[a];a=top[x]; 165 } 166 if(dep[x]>dep[y])swap(x,y); 167 if(x!=y){ 168 if(k==0) sumn+=su(1,pos[x]+1,pos[y]); 169 else if(k==1) mxn=max(mx(1,pos[x]+1,pos[y]),mxn); 170 else minn=min(mi(1,pos[x]+1,pos[y]),minn); 171 } 172 if(k==0) printf("%d ",sumn); 173 else if(k==1) printf("%d ",mxn); 174 else printf("%d ",minn); 175 } 176 int main(){ 177 //freopen("wtf.in","r",stdin); 178 //freopen("wtf.out","w",stdout); 179 scanf("%d",&n);int x,y,z; 180 for(int i=1;i<n;i++){ 181 scanf("%d%d%d",&x,&y,&z); 182 x++;y++; 183 init(x,y,z);init(y,x,z); 184 }tot=0;dfs1(1,1);dfs2(1,1); 185 build(1,1,n); 186 scanf("%d",&m); 187 for(int i=1;i<=m;i++){ 188 scanf("%s%d%d",&ch,&x,&y); 189 if(ch[0]=='N') mrev(x+1,y+1); 190 else if(ch[0]=='C')cov(1,pos[crs[x]],pos[crs[x]],y); 191 else if(ch[1]=='U')doit(x+1,y+1,0); 192 else if(ch[1]=='A')doit(x+1,y+1,1); 193 else if(ch[1]=='I')doit(x+1,y+1,2); 194 } 195 return 0; 196 }