【题目描述】
Ray 乐忠于旅游,这次他来到了T 城。T 城是一个水上城市,一共有 N 个景点,有些景点之间会用一座桥连接。为了方便游客到达每个景点但又为了节约成本,T 城的任意两个景点之间有且只有一条路径。换句话说, T 城中只有N ? 1 座桥。Ray 发现,有些桥上可以看到美丽的景色,让人心情愉悦,但有些桥狭窄泥泞,令人烦躁。于是,他给每座桥定义一个愉悦度w,也就是说,Ray 经过这座桥会增加w 的愉悦度,这或许是正的也可能是负的。有时,Ray 看待同一座桥的心情也会发生改变。现在,Ray 想让你帮他计算从u 景点到v 景点能获得的总愉悦度。有时,他还想知道某段路上最美丽的桥所提供的最大愉悦度,或是某段路上最糟糕的一座桥提供的最低愉悦度。
【输入格式】
输入的第一行包含一个整数N,表示T 城中的景点个数。景点编号为 0...N ? 1。
接下来N ? 1 行,每行三个整数u、v 和w,表示有一条u 到v,使 Ray 愉悦度增加w 的桥。桥的编号为1...N ? 1。|w| <= 1000。
输入的第N + 1 行包含一个整数M,表示Ray 的操作数目。
接下来有M 行,每行描述了一个操作,操作有如下五种形式: C i w,表示Ray 对于经过第i 座桥的愉悦度变成了w。 N u v,表示Ray 对于经过景点u 到v 的路径上的每一座桥的愉悦度都变成原来的相反数。 SUM u v,表示询问从景点u 到v 所获得的总愉悦度。 MAX u v,表示询问从景点u 到v 的路径上的所有桥中某一座桥所提供的最大愉悦度。 MIN u v,表示询问从景点u 到v 的路径上的所有桥中某一座桥所提供的最小愉悦度。测试数据保证,任意时刻,Ray 对于经过每一座桥的愉悦度的绝对值小于等于1000。
【输出格式】
对于每一个询问(操作S、MAX 和MIN),输出答案。
恶心的树剖,支持区间取反,单点修改,区间最大最小,区间和的查询的的边权树链剖分
单点修改操作跟普通树剖一样
区间取反就通过最大值与最小值相互转化,打好标记即可
#pragma GCC optimize(2)
#include<iostream>
#include<cstdio>
#include<queue>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<bitset>
using namespace std;
int ans,n,h[30005],cnt,prt[30005],deep[30005],size[30005],son[30005],top[30005],seg[30005],rev[30005],a[30005],tag[80005],m;
struct Tree
{
int maxx,minn,sum;
}tree[120005];
struct edge
{
int to,next,v;
}w[60005];
struct que
{
int x,y,v;
}e[60005];
void add(int x,int y,int z)
{
cnt++;
w[cnt].to=y;
w[cnt].v=z;
w[cnt].next=h[x];
h[x]=cnt;
}
void dfs(int x,int fa)
{
size[x]=1;
prt[x]=fa;
deep[x]=deep[fa]+1;
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(to==fa) continue;
dfs(to,x);
a[to]=w[i].v;
size[x]+=size[to];
if(size[to]>size[son[x]]) son[x]=to;
}
}
void dfs2(int x,int fa)
{
seg[x]=++seg[0];
top[x]=fa;
if(son[x]) dfs2(son[x],top[x]);
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(top[to]) continue;
dfs2(to,to);
}
}
void pushdown(int k)
{
int maxx=tree[k<<1].maxx;
tree[k<<1].maxx=-tree[k<<1].minn;
tree[k<<1].minn=-maxx;
tree[k<<1].sum=-tree[k<<1].sum;
tag[k<<1]=tag[k]-tag[k<<1];
maxx=tree[k<<1|1].maxx;
tree[k<<1|1].maxx=-tree[k<<1|1].minn;
tree[k<<1|1].minn=-maxx;
tree[k<<1|1].sum=-tree[k<<1|1].sum;
tag[k<<1|1]=tag[k]-tag[k<<1|1];
tag[k]=0;
}
void push_up(int k)
{
tree[k].maxx=max(tree[k<<1].maxx,tree[k<<1|1].maxx);
tree[k].minn=min(tree[k<<1].minn,tree[k<<1|1].minn);
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void update(int k,int l,int r,int ql,int qr)
{
// cout<<k<<" "<<l<<" "<<r<<" "<<ql<<" "<<qr<<endl;
if(ql<=l&&r<=qr)
{
tag[k]=1-tag[k];
int maxx=tree[k].maxx;
tree[k].maxx=-tree[k].minn;
tree[k].minn=-maxx;
tree[k].sum=-tree[k].sum;
return ;
}
if(tag[k]) pushdown(k);
int mid=(l+r)>>1;
if(ql<=mid) update(k<<1,l,mid,ql,qr);
if(mid<qr) update(k<<1|1,mid+1,r,ql,qr);
push_up(k);
}
void uprange(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]]) swap(x,y);
update(1,1,seg[0],seg[top[x]],seg[x]);
x=prt[top[x]];
}
if(deep[x]>deep[y]) swap(x,y);
update(1,1,seg[0],seg[son[x]],seg[y]);
}
void change(int k,int l,int r,int x,int v)
{
if(l==r)
{
tree[k].maxx=tree[k].minn=tree[k].sum=v;
return ;
}
int mid=(l+r)>>1;
if(tag[k]) pushdown(k);
if(x<=mid) change(k<<1,l,mid,x,v);
else change(k<<1|1,mid+1,r,x,v);
push_up(k);
}
void query(int k,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)
{
ans=max(ans,tree[k].maxx);
return ;
}
int mid=(l+r)>>1;
if(tag[k]) pushdown(k);
if(ql<=mid) query(k<<1,l,mid,ql,qr);
if(mid<qr) query(k<<1|1,mid+1,r,ql,qr);
push_up(k);
}
void query2(int k,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)
{
ans=min(ans,tree[k].minn);
return ;
}
int mid=(l+r)>>1;
if(tag[k]) pushdown(k);
if(ql<=mid) query2(k<<1,l,mid,ql,qr);
if(mid<qr) query2(k<<1|1,mid+1,r,ql,qr);
push_up(k);
}
void query3(int k,int l,int r,int ql,int qr)
{
if(ql<=l&&r<=qr)
{
ans+=tree[k].sum;
return ;
}
int mid=(l+r)>>1;
if(tag[k]) pushdown(k);
if(ql<=mid) query3(k<<1,l,mid,ql,qr);
if(mid<qr) query3(k<<1|1,mid+1,r,ql,qr);
push_up(k);
}
void ask(int x,int y)
{
ans=-0x3f3f3f3f;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]]) swap(x,y);
query(1,1,seg[0],seg[top[x]],seg[x]);
x=prt[top[x]];
}
if(deep[x]>deep[y]) swap(x,y);
// cout<<son[x]<<" "<<seg[y]<<endl;
query(1,1,seg[0],seg[son[x]],seg[y]);
}
void ask2(int x,int y)
{
ans=0x3f3f3f3f;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]]) swap(x,y);
query2(1,1,seg[0],seg[top[x]],seg[x]);
x=prt[top[x]];
}
if(deep[x]>deep[y]) swap(x,y);
// cout<<son[x]<<" "<<seg[y]<<endl;
query2(1,1,seg[0],seg[son[x]],seg[y]);
}
void ask3(int x,int y)
{
ans=0;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]]) swap(x,y);
query3(1,1,seg[0],seg[top[x]],seg[x]);
x=prt[top[x]];
}
if(deep[x]>deep[y]) swap(x,y);
// cout<<son[x]<<" "<<seg[y]<<endl;
query3(1,1,seg[0],seg[son[x]],seg[y]);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
x++,y++;
add(x,y,z);
add(y,x,z);
e[i].x=x,e[i].y=y,e[i].v=z;
}
dfs(1,0);
dfs2(1,1);
for(int i=1;i<n;i++)
{
int nx=e[i].x,ny=e[i].y;
if(deep[nx]>deep[ny]) swap(nx,ny);
change(1,1,seg[0],seg[ny],e[i].v);
}
char s[10];
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%s",s);
int x,y;
scanf("%d%d",&x,&y);
x++,y++;
if(s[0]=='C')
{
x--,y--;
int nx=e[x].x,ny=e[x].y;
if(prt[nx]==ny) change(1,1,seg[0],seg[nx],y);
else change(1,1,seg[0],seg[ny],y);
}
else if(s[0]=='N')
{
uprange(x,y);
}
else if(s[1]=='I')
{
ask2(x,y);
printf("%d\n",ans);
}
else if(s[1]=='U')
{
ask3(x,y);
printf("%d\n",ans);
}
else
{
ask(x,y);
printf("%d\n",ans);
}
}
}