BZOJ 3159: 决战

1 sec 512MB

题意：

给你一颗\(n\)个点，初始点权为\(0\)的有跟树，要求支持

• Increase x y w 将路径\(x\)到\(y\)所有点点权加上\(w\)
• Sum x y 询问路径\(x\)到\(y\)的点权和
• Major x y 询问从路径\(x\)到\(y\)最大点权
• Minor x y 询问最小点权
• Invert x y 将路径上的点权翻转

有个性质，修改操作一定满足\(x\)为\(y\)祖先或者\(y\)为\(x\)祖先（实际没什么用处）

\(n\le 50000,|w|\le 1000\)

输入格式

第一行有三个整数\(N\)、\(M\)和\(R\)，分别表示树的节点数、指令和询问总数，以及树的跟。

接下来\(N-1\)行，每行两个整数\(u\)和\(v\)，表示一条边。

接下来\(M\)行，每行描述一个指令或询问，格式见题意描述。

输出格式

对于每个询问操作，输出所求的值。

老年菜鸡选手实在写不动大数据结构题...

有两个做法，暂时只写了一种，有可能一会儿会把另一种补充一下。

考虑树链剖分，但是不能维护翻转

考虑平衡树可以维护翻转

于是我们可以拿平衡树维护树剖的每条链

然后每次修改只有\(\log\)条链，我们把每条链要修改的地方拎出来，然后塞到一个平衡树里面，打一个翻转tag，再塞回去就可以了

用fhqtreap实现起来比较方便

说起来蛮简单，写起来还是蛮难受的的

Code:

#include <cstdio>
#include <cctype>
#include <cstdlib>
#include <algorithm>
#define ll long long
const int SIZE=1<<21;
char ibuf[SIZE],*iS,*iT;
//#define gc() (iS==iT?(iT=(iS=ibuf)+fread(ibuf,1,SIZE,stdin),iS==iT?EOF:*iS++):*iS++)
#define gc() getchar()
template <class T>
void read(T &x)
{int f=0;x=0;char c=gc();while(!isdigit(c)) f|=c=='-',c=gc();while(isdigit(c)) x=x*10+c-'0',c=gc();if(f) x=-x;
}
void reads(char *s)
{int k=0;char c=gc();while(c<'A'||c>'Z') c=gc();while((c>='a'&&c<='z')||(c>='A'&&c<='Z')) s[k++]=c,c=gc();
}
const int N=5e4+10;
inline void ckmax(int &x,int y){x=x>y?x:y;}
inline void ckmin(int &x,int y){x=x<y?x:y;}
int n,m,r,root[N];
namespace treap
{int ch[N][2],mx[N],mi[N],siz[N],val[N],dat[N],tag[N],reve[N],tot;ll sum[N];#define ls ch[now][0]#define rs ch[now][1]void updata(int now){sum[now]=sum[ls]+dat[now]+sum[rs];siz[now]=siz[ls]+1+siz[rs];mx[now]=mi[now]=dat[now];if(ls){ckmax(mx[now],mx[ls]);ckmin(mi[now],mi[ls]);}if(rs){ckmax(mx[now],mx[rs]);ckmin(mi[now],mi[rs]);}}void Reverse(int now){std::swap(ls,rs),reve[now]^=1;}void upt(int now,int d){dat[now]+=d;tag[now]+=d;mi[now]+=d;mx[now]+=d;sum[now]+=d*siz[now];}void pushdown(int now){if(reve[now]){if(ls) Reverse(ls);if(rs) Reverse(rs);reve[now]=0;}if(tag[now]){if(ls) upt(ls,tag[now]);if(rs) upt(rs,tag[now]);tag[now]=0;}}void split(int now,int &x,int &y,int k){if(!now){x=y=0;return;}pushdown(now);if(k<=siz[ls])y=now,split(ls,x,ch[y][0],k);elsex=now,split(rs,ch[x][1],y,k-siz[ls]-1);updata(now);}int Merge(int x,int y){if(!x||!y) return x^y;pushdown(x),pushdown(y);if(val[x]<val[y]){ch[x][1]=Merge(ch[x][1],y);updata(x);return x;}else{ch[y][0]=Merge(x,ch[y][0]);updata(y);return y;}}int New(int d){val[++tot]=rand(),siz[tot]=1,sum[tot]=mi[tot]=mx[tot]=dat[tot]=d;return tot;}void ins(int id,int d){root[id]=Merge(root[id],New(d));}ll querysum(int id,int l,int r){int x,y,z;split(root[id],x,y,r);split(x,x,z,l-1);ll ret=sum[z];root[id]=Merge(x,Merge(z,y));return ret;}int querymi(int id,int l,int r){int x,y,z;split(root[id],x,y,r);split(x,x,z,l-1);int ret=mi[z];root[id]=Merge(x,Merge(z,y));return ret;}int querymx(int id,int l,int r){int x,y,z;split(root[id],x,y,r);split(x,x,z,l-1);int ret=mx[z];root[id]=Merge(x,Merge(z,y));return ret;}void modify(int id,int l,int r,int w){int x,y,z;split(root[id],x,y,r);split(x,x,z,l-1);upt(z,w);root[id]=Merge(x,Merge(z,y));}
}
using treap::modify;
using treap::querysum;
using treap::querymi;
using treap::querymx;
using treap::ins;
using treap::Merge;
using treap::split;
using treap::Reverse;
int head[N],to[N<<1],Next[N<<1],cnt;
void add(int u,int v)
{to[++cnt]=v,Next[cnt]=head[u],head[u]=cnt;
}
int top[N],ws[N],siz[N],par[N],dep[N],rk[N],num[N];
void dfs(int now)
{siz[now]=1;dep[now]=dep[par[now]]+1;for(int v,i=head[now];i;i=Next[i])if((v=to[i])!=par[now]){par[v]=now;dfs(v);siz[now]+=siz[v];if(siz[ws[now]]<siz[v]) ws[now]=v;}
}
int lin,tot[N];
void dfs(int now,int id,int anc)
{top[now]=anc;num[now]=id;rk[now]=++tot[id];ins(id,0);if(ws[now]) dfs(ws[now],id,anc);for(int v,i=head[now];i;i=Next[i])if(!num[v=to[i]])dfs(v,++lin,v);
}
void modi(int x,int y,int w)
{if(dep[x]<dep[y]) std::swap(x,y);while(top[x]!=top[y]){modify(num[x],1,rk[x],w);x=par[top[x]];}if(dep[x]<dep[y]) std::swap(x,y);modify(num[x],rk[y],rk[x],w);
}
void qrysum(int x,int y)
{ll ret=0;while(top[x]!=top[y]){if(dep[top[x]]>dep[top[y]]){ret+=querysum(num[x],1,rk[x]);x=par[top[x]];}else{ret+=querysum(num[y],1,rk[y]);y=par[top[y]];}}if(dep[x]<dep[y]) std::swap(x,y);ret+=querysum(num[x],rk[y],rk[x]);printf("%lld\n",ret);
}
void qrymx(int x,int y)
{int ret=-(1<<30);while(top[x]!=top[y]){if(dep[top[x]]>dep[top[y]]){ckmax(ret,querymx(num[x],1,rk[x]));x=par[top[x]];}else{ckmax(ret,querymx(num[y],1,rk[y]));y=par[top[y]];}}if(dep[x]<dep[y]) std::swap(x,y);ckmax(ret,querymx(num[x],rk[y],rk[x]));printf("%d\n",ret);
}
void qrymi(int x,int y)
{int ret=1<<30;while(top[x]!=top[y]){if(dep[top[x]]>dep[top[y]]){ckmin(ret,querymi(num[x],1,rk[x]));x=par[top[x]];}else{ckmin(ret,querymi(num[y],1,rk[y]));y=par[top[y]];}}if(dep[x]<dep[y]) std::swap(x,y);ckmin(ret,querymi(num[x],rk[y],rk[x]));printf("%d\n",ret);
}
void rev(int x,int y)
{int tx=x,ty=y;if(dep[x]<dep[y]) std::swap(x,y);int rt=0;while(top[x]!=top[y]){int id=num[x],k=rk[x],pre;split(root[id],pre,root[id],k);rt=Merge(pre,rt);x=par[top[x]];}if(dep[x]<dep[y]) std::swap(x,y);int id=num[x],l=rk[y],r=rk[x],a,b,c;split(root[id],a,c,r);split(a,a,b,l-1);rt=Merge(b,rt);Reverse(rt);x=tx,y=ty;if(dep[x]<dep[y]) std::swap(x,y);while(top[x]!=top[y]){int id=num[x],k=rk[x],pre;split(rt,rt,pre,treap::siz[rt]-k);root[id]=Merge(pre,root[id]);x=par[top[x]];}id=num[x];root[id]=Merge(a,Merge(rt,c));
}
int main()
{//freopen("data.in","r",stdin);//freopen("data.out","w",stdout);read(n),read(m),read(r);for(int u,v,i=1;i<n;i++) read(u),read(v),add(u,v),add(v,u);dfs(r);dfs(r,++lin,r);char op[23];for(int x,y,w,i=1;i<=m;i++){reads(op),read(x),read(y);if(op[0]=='I'){if(op[2]=='c') read(w),modi(x,y,w);else rev(x,y);}else if(op[0]=='S') qrysum(x,y);else{if(op[1]=='a') qrymx(x,y);else qrymi(x,y);}}return 0;
}

2019.5.22