luogu P1505 [国家集训队]旅游
题面传送门
第一次一遍过国集紫题。好激动。
一看就是树剖，只不过操作有点多。
把每条边的权值放在儿子节点即可。
对于取相反数放懒标记即可。
其他是树剖正常操作。
代码实现:
#include<cstdio>
#include<cstring>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,sx,sy,sz,x[200039],y[200039],z[200039],son[200039],siz[200039],d[200039],top[200039],id[200039],idea,fa[200039],flag[800039],smax[800039],smin[800039],sum[800039];
char _s;
struct yyy{int to,w,z;};
struct ljb{
int head,h[200039];
yyy f[400039];
inline void add(int x,int y,int z){
f[++head]=(yyy){y,z,h[x]};
h[x]=head;
}
}s;
inline void swap(int &x,int &y){x^=y,y^=x,x^=y;}
inline void push(int l,int r,int now){
flag[now<<1]^=1;
flag[now<<1|1]^=1;
swap(smax[now<<1],smin[now<<1]);
smax[now<<1]*=-1;smin[now<<1]*=-1;
swap(smax[now<<1|1],smin[now<<1|1]);
smax[now<<1|1]*=-1;smin[now<<1|1]*=-1;
sum[now<<1]*=-1;
sum[now<<1|1]*=-1;
flag[now]=0;
}
inline void hebing(int now){
sum[now]=sum[now<<1]+sum[now<<1|1];
smax[now]=max(smax[now<<1],smax[now<<1|1]);
smin[now]=min(smin[now<<1],smin[now<<1|1]);
}
inline void get1(int x,int z,int l,int r,int now){
if(l==r) {smax[now]=smin[now]=sum[now]=z;return;}
if(flag[now]) push(l,r,now);
int m=(l+r)>>1;
if(x<=m) get1(x,z,l,m,now<<1);
else get1(x,z,m+1,r,now<<1|1);
hebing(now);
return;
}
inline void get2(int x,int y,int l,int r,int now){
if(x<=l&&r<=y){flag[now]^=1;sum[now]*=-1;swap(smax[now],smin[now]);smax[now]*=-1;smin[now]*=-1;return;}
if(flag[now]) push(l,r,now);
int m=(l+r)>>1;
if(x<=m) get2(x,y,l,m,now<<1);
if(y>m) get2(x,y,m+1,r,now<<1|1);
hebing(now);
}
inline int find1(int x,int y,int l,int r,int now){
if(x<=l&&r<=y) return sum[now];
if(flag[now]) push(l,r,now);
int m=(l+r)>>1,fs=0;
if(x<=m) fs+=find1(x,y,l,m,now<<1);
if(y>m) fs+=find1(x,y,m+1,r,now<<1|1);
return fs;
}
inline int find2(int x,int y,int l,int r,int now){
if(x<=l&&r<=y) return smax[now];
if(flag[now]) push(l,r,now);
int m=(l+r)>>1,ans1=-1e9,ans2=-1e9;
if(x<=m) ans1=find2(x,y,l,m,now<<1);
if(y>m) ans2=find2(x,y,m+1,r,now<<1|1);
return max(ans1,ans2);
}
inline int find3(int x,int y,int l,int r,int now){
if(x<=l&&r<=y) return smin[now];
if(flag[now]) push(l,r,now);
int m=(l+r)>>1,ans1=1e9,ans2=1e9;
if(x<=m) ans1=find3(x,y,l,m,now<<1);
if(y>m) ans2=find3(x,y,m+1,r,now<<1|1);
return min(ans1,ans2);
}
inline void dfs1(int x,int last){
d[x]=d[last]+1;
fa[x]=last;
siz[x]=1;
int cur=s.h[x],pus=-1;
yyy tmp;
while(cur!=-1){
tmp=s.f[cur];
if(tmp.to!=last){
dfs1(tmp.to,x);
siz[x]+=siz[tmp.to];
if(pus<siz[tmp.to]) pus=siz[tmp.to],son[x]=tmp.to;
}
cur=tmp.z;
}
}
inline void dfs2(int x,int last){
top[x]=last;
id[x]=++idea;
if(!son[x]) return;
int cur=s.h[x];
yyy tmp;
while(cur!=-1){
tmp=s.f[cur];
if(tmp.to==son[x]) get1(idea+1,tmp.w,1,n,1),dfs2(tmp.to,last);
cur=tmp.z;
}
cur=s.h[x];
while(cur!=-1){
tmp=s.f[cur];
if(tmp.to!=son[x]&&tmp.to!=fa[x]) get1(idea+1,tmp.w,1,n,1),dfs2(tmp.to,tmp.to);
cur=tmp.z;
}
}
inline void gets1(int x,int y,int z){
if(d[x]>d[y]) swap(x,y);
get1(id[y],z,1,n,1);
}
inline void gets2(int x,int y){
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]]) swap(x,y);
get2(id[top[x]],id[x],1,n,1);
x=fa[top[x]];
}
if(d[x]>d[y]) swap(x,y);
get2(id[x]+1,id[y],1,n,1);
}
inline int finds1(int x,int y){
int ans=0;
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]]) swap(x,y);
ans+=find1(id[top[x]],id[x],1,n,1);
x=fa[top[x]];
}
if(d[x]>d[y]) swap(x,y);
return ans+find1(id[x]+1,id[y],1,n,1);
}
inline int finds2(int x,int y){
int ans=-1e9,tot=0;
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]]) swap(x,y);
tot=find2(id[top[x]],id[x],1,n,1);
ans=max(ans,tot);
x=fa[top[x]];
}
if(d[x]>d[y]) swap(x,y);
tot=find2(id[x]+1,id[y],1,n,1);
return max(ans,tot);
}
inline int finds3(int x,int y){
int ans=1e9,tot=0;
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]]) swap(x,y);
tot=find3(id[top[x]],id[x],1,n,1);
ans=min(ans,tot);
x=fa[top[x]];
}
if(d[x]>d[y]) swap(x,y);
tot=find3(id[x]+1,id[y],1,n,1);
return min(ans,tot);
}
int main(){
memset(s.h,-1,sizeof(s.h));
register int i;
scanf("%d",&n);
for(i=1;i<n;i++) scanf("%d%d%d",&x[i],&y[i],&z[i]),x[i]++,y[i]++,s.add(x[i],y[i],z[i]),s.add(y[i],x[i],z[i]);
dfs1(1,0);
dfs2(1,1);
scanf("%d",&m);
for(i=1;i<=m;i++){
_s=getchar();
while(_s<'A'||_s>'Z') _s=getchar();
if(_s=='S'){
_s=getchar();_s=getchar();
scanf("%d%d",&sx,&sy);
sx++;sy++;
printf("%d\n",finds1(sx,sy));
}
else if(_s=='M'){
_s=getchar();
if(_s=='A'){
_s=getchar();
scanf("%d%d",&sx,&sy);
sx++;sy++;
printf("%d\n",finds2(sx,sy));
}
else{
_s=getchar();
scanf("%d%d",&sx,&sy);
sx++;sy++;
printf("%d\n",finds3(sx,sy));
}
}
else if(_s=='C'){
scanf("%d%d",&sx,&sz);
gets1(x[sx],y[sx],sz);
}
else{
scanf("%d%d",&sx,&sy);
sx++;sy++;
gets2(sx,sy);
}
}
}
不得不说这道题\(5k\)代码真是毒瘤。
