BZOJ2157: 旅游 树链剖分 线段树

 
在对树中数据进行改动的时候需要很多pushdown(具体操作见代码),不然会wa,大概原因和线段树区间修改需要很多pushup是一样的。
这个轻重链的方法特别好用,虽然第一次写树链剖分但是容易理解又有优秀复杂度的结构让人情不自禁orz。
(后来发现很久以前学lca的时候就学了树链剖分只不过忘了,mdzz)
因为忘了去掉测试代码的freopen,re了4次(虽然有三次就算不re也wa),发现一行一行检查对于数据结构的调试很有用(虽然一直找不出来re原因但是我起码避免了很多wa呀)。
 
代码
 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=;
const long long modn=;
const int minf=<<;
int n,m;char ch[]={};
struct nod{
int y,next,v;
}e[maxn*];
int head[maxn]={},tot=;
struct seg{
int mi,mx,sum,c,l,r;
int f,rev;
seg(){mi=mx=sum=c=l=r=;f=rev=;}
}t[maxn*];
int dep[maxn]={},crs[maxn]={},pos[maxn]={},road[maxn]={};
int kid[maxn]={},fa[maxn]={},top[maxn]={};
void init(int x,int y,int v){
e[++tot].v=v;e[tot].y=y;
e[tot].next=head[x];head[x]=tot;
}
int dfs1(int x,int pa){
int y,siz=,tsn=,ma=;
fa[x]=pa;dep[x]=dep[pa]+;
for(int i=head[x];i;i=e[i].next){
y=e[i].y;
if(y==pa)continue;
siz=dfs1(y,x);tsn+=siz;
if(siz>ma)ma=siz,kid[x]=y;
crs[(i+)/]=y;
}
return tsn;
}
void dfs2(int x,int pa){
int y,v;
pos[x]=++tot;top[x]=pa;
if(kid[x])dfs2(kid[x],pa);
for(int i=head[x];i;i=e[i].next){
y=e[i].y;v=e[i].v;
if(y==kid[x])road[pos[kid[x]]]=v;
if(y==kid[x]||y==fa[x])continue;
road[tot+]=v;
dfs2(y,y);
}
}
void pushup(int x){
int l=x*,r=x*+;
t[x].mx=max(t[l].mx,t[r].mx);
t[x].mi=min(t[l].mi,t[r].mi);
t[x].sum=t[l].sum+t[r].sum;
}
void pushdown(int x){
if(t[x].f){
int l=x*,r=x*+;
t[x].rev=t[x].f=;
t[x].mx=t[x].mi=t[x].c;
t[x].sum=t[x].c*(t[x].r-t[x].l+);
if(t[x].l<t[x].r){
t[l].c=t[r].c=t[x].c;
t[l].f=t[r].f=;
t[l].rev=t[r].rev=;
}
return;
}
if(t[x].rev){
int l=x*,r=x*+;
t[x].rev=;
swap(t[x].mi,t[x].mx);
t[x].mx*=-;t[x].mi*=-;t[x].sum*=-;
if(t[x].l<t[x].r){
t[l].c*=-;t[l].rev^=;
t[r].c*=-;t[r].rev^=;
}
}
}
void build(int x,int l,int r){
t[x].l=l,t[x].r=r;
if(l==r){
t[x].mx=t[x].mi=t[x].sum=road[l];
return;
}
int mid=(l+r)/;
build(x*,l,mid);
build(x*+,mid+,r);
pushup(x);
}
void cov(int x,int l,int r,int w){
if(t[x].l>=l&&t[x].r<=r){
t[x].f=;t[x].c=w; pushdown(x);
return;
}
pushdown(x);
int mid=(t[x].l+t[x].r)/;
if(l<=mid)cov(x*,l,r,w);
pushdown(x*);
if(r>mid) cov(x*+,l,r,w);
pushdown(x*+);
pushup(x);
}
void revs(int x,int l,int r){
if(t[x].l>=l&&t[x].r<=r){
t[x].c*=-;t[x].rev^=;
pushdown(x);
return;
}pushdown(x);
int mid=(t[x].l+t[x].r)/;
if(l<=mid) revs(x*,l,r);
pushdown(x*);
if(r>mid) revs(x*+,l,r);
pushdown(x*+);
pushup(x);
}
int mi(int x,int l,int r){
pushdown(x);
if(t[x].l>=l&&t[x].r<=r){
return t[x].mi;
}
int mid=(t[x].l+t[x].r)/,minn=minf;
if(l<=mid) minn=min(minn,mi(x*,l,r));
if(r>mid) minn=min(minn,mi(x*+,l,r));
return minn;
}
int mx(int x,int l,int r){
pushdown(x);
if(t[x].l>=l&&t[x].r<=r){
return t[x].mx;
}
int mid=(t[x].l+t[x].r)/,mxn=-minf;
if(l<=mid) mxn=max(mxn,mx(x*,l,r));
if(r>mid) mxn=max(mxn,mx(x*+,l,r));
return mxn;
}
int su(int x,int l,int r){
pushdown(x);
if(t[x].l>=l&&t[x].r<=r){
return t[x].sum;
}
int mid=(t[x].l+t[x].r)/,sumn=;
if(l<=mid) sumn+=su(x*,l,r);
if(r>mid) sumn+=su(x*+,l,r);
return sumn;
}
void mrev(int x,int y){
int a,b;
for(a=top[x],b=top[y];a!=b;){
if(dep[a]<dep[b])swap(a,b),swap(x,y);
revs(,pos[a],pos[x]);
x=fa[a];a=top[x];
}
if(dep[x]>dep[y])swap(x,y);
if(x!=y) revs(,pos[x]+,pos[y]);
}
void doit(int x,int y,int k){
int a,b,mxn=-minf,minn=minf,sumn=;
for(a=top[x],b=top[y];a!=b;){
if(dep[a]<dep[b])swap(a,b),swap(x,y);
if(k==) sumn+=su(,pos[a],pos[x]);
else if(k==) mxn=max(mx(,pos[a],pos[x]),mxn);
else minn=min(mi(,pos[a],pos[x]),minn);
x=fa[a];a=top[x];
}
if(dep[x]>dep[y])swap(x,y);
if(x!=y){
if(k==) sumn+=su(,pos[x]+,pos[y]);
else if(k==) mxn=max(mx(,pos[x]+,pos[y]),mxn);
else minn=min(mi(,pos[x]+,pos[y]),minn);
}
if(k==) printf("%d\n",sumn);
else if(k==) printf("%d\n",mxn);
else printf("%d\n",minn);
}
int main(){
//freopen("wtf.in","r",stdin);
//freopen("wtf.out","w",stdout);
scanf("%d",&n);int x,y,z;
for(int i=;i<n;i++){
scanf("%d%d%d",&x,&y,&z);
x++;y++;
init(x,y,z);init(y,x,z);
}tot=;dfs1(,);dfs2(,);
build(,,n);
scanf("%d",&m);
for(int i=;i<=m;i++){
scanf("%s%d%d",&ch,&x,&y);
if(ch[]=='N') mrev(x+,y+);
else if(ch[]=='C')cov(,pos[crs[x]],pos[crs[x]],y);
else if(ch[]=='U')doit(x+,y+,);
else if(ch[]=='A')doit(x+,y+,);
else if(ch[]=='I')doit(x+,y+,);
}
return ;
}
上一篇:并查集+树链剖分+线段树 HDOJ 5458 Stability(稳定性)


下一篇:BZOJ 1984: 月下“毛景树” [树链剖分 边权]