BZOJ 2157: 旅游
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
const int M=200005;
const int inf=1e9;
struct Tree{int l,r,opp,mx,mn,sum;} T[N*4];
int n,ff[N],tt[N],ww[N];
int cnt,to[N],nxt[N],lj[N],a[N];
void insert(int f,int t)
{
cnt++,to[cnt]=t,nxt[cnt]=lj[f],lj[f]=cnt;
cnt++,to[cnt]=f,nxt[cnt]=lj[t],lj[t]=cnt;
}
void pushdown(int x)
{
int lc=x<<1,rc=x<<1|1,tmp;
tmp=T[lc].mx;
T[lc].mx=-T[lc].mn;
T[lc].mn=-tmp;
T[lc].sum=-T[lc].sum;
T[lc].opp^=1;
tmp=T[rc].mx;
T[rc].mx=-T[rc].mn;
T[rc].mn=-tmp;
T[rc].sum=-T[rc].sum;
T[rc].opp^=1;
T[x].opp=0;
}
void pushup(int x)
{
T[x].mx=max(T[x<<1].mx,T[x<<1|1].mx);
T[x].mn=min(T[x<<1].mn,T[x<<1|1].mn);
T[x].sum=T[x<<1].sum+T[x<<1|1].sum;
}
int fa[N],pos[N],npos[N],son[N],s[N],dep[N],top[N],dian[N];
void dfs1(int x)
{
dep[x]=dep[fa[x]]+1;
s[x]=1;
for(int i=lj[x];i;i=nxt[i])
if(to[i]!=fa[x])
{
fa[to[i]]=x;
dfs1(to[i]);
s[x]+=s[to[i]];
if(s[to[i]]>s[son[x]]) son[x]=to[i];
}
}
void dfs2(int x)
{
pos[x]=++cnt;
npos[cnt]=x;
if(son[fa[x]]==x) top[x]=top[fa[x]];
else top[x]=x;
if(son[x]) dfs2(son[x]);
for(int i=lj[x];i;i=nxt[i])
if(to[i]!=son[x]&&to[i]!=fa[x]) dfs2(to[i]);
}
void build(int x,int l,int r)
{
T[x].l=l,T[x].r=r;
if(l==r)
{
T[x].mx=a[npos[l]];
T[x].mn=a[npos[l]];
T[x].sum=a[npos[l]];
return;
}
int mid=(l+r)>>1;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
pushup(x);
}
void change(int x,int k,int v)
{
if(T[x].l==T[x].r)
{
T[x].mx=v;
T[x].mn=v;
T[x].sum=v;
return;
}
int mid=(T[x].l+T[x].r)>>1;
if(T[x].opp) pushdown(x);
if(k<=mid) change(x<<1,k,v);
else change(x<<1|1,k,v);
pushup(x);
}
int Querymax(int x,int l,int r)
{
if(T[x].l==l&&T[x].r==r) return T[x].mx;
if(T[x].opp) pushdown(x);
int mid=(T[x].l+T[x].r)>>1;
if(r<=mid) return Querymax(x<<1,l,r);
else if(l>mid) return Querymax(x<<1|1,l,r);
else return max(Querymax(x<<1,l,mid),Querymax(x<<1|1,mid+1,r));
}
int Querymin(int x,int l,int r)
{
if(T[x].l==l&&T[x].r==r) return T[x].mn;
if(T[x].opp) pushdown(x);
int mid=(T[x].l+T[x].r)>>1;
if(r<=mid) return Querymin(x<<1,l,r);
else if(l>mid) return Querymin(x<<1|1,l,r);
else return min(Querymin(x<<1,l,mid),Querymin(x<<1|1,mid+1,r));
}
int Querysum(int x,int l,int r)
{
if(T[x].l==l&&T[x].r==r) return T[x].sum;
if(T[x].opp) pushdown(x);
int mid=(T[x].l+T[x].r)>>1;
if(r<=mid) return Querysum(x<<1,l,r);
else if(l>mid) return Querysum(x<<1|1,l,r);
else return Querysum(x<<1,l,mid)+Querysum(x<<1|1,mid+1,r);
}
void Modify(int x,int l,int r)
{
if(T[x].l==l&&T[x].r==r)
{
int tmp=T[x].mx;
T[x].mx=-T[x].mn;
T[x].mn=-tmp;
T[x].sum=-T[x].sum;
T[x].opp^=1;
return;
}
if(T[x].opp) pushdown(x);
int mid=(T[x].l+T[x].r)>>1;
if(r<=mid) Modify(x<<1,l,r);
else if(l>mid) Modify(x<<1|1,l,r);
else Modify(x<<1,l,mid),Modify(x<<1|1,mid+1,r);
pushup(x);
}
int mx,mn,tot;
void Get(int x,int y,int o)
{
int fx=top[x],fy=top[y];
mx=-inf,mn=inf,tot=0;
while(fx!=fy)
{
if(dep[fx]<dep[fy]) swap(x,y),swap(fx,fy);
if(o==1) mx=max(mx,Querymax(1,pos[fx],pos[x]));
if(o==2) mn=min(mn,Querymin(1,pos[fx],pos[x]));
if(o==3) Modify(1,pos[fx],pos[x]);
if(o==4) tot+=Querysum(1,pos[fx],pos[x]);
x=fa[fx];
fx=top[x];
}
if(dep[x]<dep[y]) swap(x,y);
if(x!=y)
{
if(o==1) mx=max(mx,Querymax(1,pos[y]+1,pos[x]));
if(o==2) mn=min(mn,Querymin(1,pos[y]+1,pos[x]));
if(o==3) Modify(1,pos[y]+1,pos[x]);
if(o==4) tot+=Querysum(1,pos[y]+1,pos[x]);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&ff[i],&tt[i],&ww[i]);
ff[i]++,tt[i]++;
insert(ff[i],tt[i]);
}
cnt=0;
dfs1(1);
dfs2(1);
for(int i=1;i<n;i++)
{
if(dep[ff[i]]<dep[tt[i]]) a[tt[i]]=ww[i],dian[i]=tt[i];
else a[ff[i]]=ww[i],dian[i]=ff[i];
}
build(1,1,n);
char c[10];
int x,y,Q;
scanf("%d",&Q);
while(Q--)
{
scanf("%s%d%d",c,&x,&y);
if(c[0]=='C') change(1,pos[dian[x]],y);
x++,y++;
if(c[0]=='M')
{
if(c[1]=='A')
{Get(x,y,1);printf("%d\n",mx);}
if(c[1]=='I')
{Get(x,y,2);printf("%d\n",mn);}
}
if(c[0]=='N') Get(x,y,3);
if(c[0]=='S')
{Get(x,y,4);printf("%d\n",tot);}
}
}
Description
Ray 乐忠于旅游,这次他来到了T 城。T 城是一个水上城市,一共有 N 个景点,有些景点之间会用一座桥连接。为了方便游客到达每个景点但又为了节约成本,T 城的任意两个景点之间有且只有一条路径。换句话说, T 城中只有N − 1 座桥。Ray 发现,有些桥上可以看到美丽的景色,让人心情愉悦,但有些桥狭窄泥泞,令人烦躁。于是,他给每座桥定义一个愉悦度w,也就是说,Ray 经过这座桥会增加w 的愉悦度,这或许是正的也可能是负的。有时,Ray 看待同一座桥的心情也会发生改变。现在,Ray 想让你帮他计算从u 景点到v 景点能获得的总愉悦度。有时,他还想知道某段路上最美丽的桥所提供的最大愉悦度,或是某段路上最糟糕的一座桥提供的最低愉悦度。
Input
输入的第一行包含一个整数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。
Output
对于每一个询问(操作S、MAX 和MIN),输出答案。
Sample Input
0 1 1
1 2 2
8
SUM 0 2
MAX 0 2
N 0 1
SUM 0 2
MIN 0 2
C 1 3
SUM 0 2
MAX 0 2
Sample Output
2
1
-1
5
3
HINT
一共有10 个数据,对于第i (1 <= i <= 10) 个数据, N = M = i * 2000。