[BZOJ2157]旅游
题面戳我
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
3
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
3
2
1
-1
5
3
HINT
一共有10 个数据,对于第i (1 <= i <= 10) 个数据, N = M = i * 2000。
sol
树链剖分,这个题我还能讲什么
细节,也没多少,认真写吧。。。
code
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 20005;
struct edge{int to,next,w;}a[N<<1];
int n,m,head[N],cnt=1,fa[N],dep[N],sz[N],son[N],top[N],dfn[N],ref[N],sum[N<<2],mx[N<<2],mn[N<<2],rev[N<<2];
char s[N];
int gi()
{
int x=0,w=1;char ch=getchar();
while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
if (ch=='-') w=0,ch=getchar();
while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
return w?x:-x;
}
void dfs1(int u,int f)
{
fa[u]=f;dep[u]=dep[f]+1;sz[u]=1;
for (int e=head[u];e;e=a[e].next)
{
int v=a[e].to;if (v==f) continue;
ref[e>>1]=v;dfs1(v,u);
sz[u]+=sz[v];if (sz[v]>sz[son[u]]) son[u]=v;
}
}
void dfs2(int u,int up)
{
top[u]=up;dfn[u]=++cnt;
if (son[u]) dfs2(son[u],up);
for (int e=head[u];e;e=a[e].next)
if (a[e].to!=fa[u]&&a[e].to!=son[u])
dfs2(a[e].to,a[e].to);
}
void pushup(int now)
{
sum[now]=sum[now<<1]+sum[now<<1|1];
mx[now]=max(mx[now<<1],mx[now<<1|1]);
mn[now]=min(mn[now<<1],mn[now<<1|1]);
}
void pushdown(int now)
{
sum[now<<1]*=-1;sum[now<<1|1]*=-1;
swap(mx[now<<1],mn[now<<1]);mx[now<<1]*=-1;mn[now<<1]*=-1;
swap(mx[now<<1|1],mn[now<<1|1]);mx[now<<1|1]*=-1;mn[now<<1|1]*=-1;
rev[now<<1]^=1;rev[now<<1|1]^=1;
rev[now]=0;
}
void Modify(int now,int l,int r,int pos,int val)
{
if (l==r) {sum[now]=mx[now]=mn[now]=val;return;}
if (rev[now]) pushdown(now);
int mid=l+r>>1;
if (pos<=mid) Modify(now<<1,l,mid,pos,val);
else Modify(now<<1|1,mid+1,r,pos,val);
pushup(now);
}
void Reverse(int now,int l,int r,int ql,int qr)
{
if (l>=ql&&r<=qr)
{
sum[now]*=-1;
swap(mx[now],mn[now]);mx[now]*=-1;mn[now]*=-1;
rev[now]^=1;return;
}
if (rev[now]) pushdown(now);
int mid=l+r>>1;
if (ql<=mid) Reverse(now<<1,l,mid,ql,qr);
if (qr>mid) Reverse(now<<1|1,mid+1,r,ql,qr);
pushup(now);
}
int Query_Sum(int now,int l,int r,int ql,int qr)
{
if (l>=ql&&r<=qr) return sum[now];
if (rev[now]) pushdown(now);
int mid=l+r>>1,s=0;
if (ql<=mid) s+=Query_Sum(now<<1,l,mid,ql,qr);
if (qr>mid) s+=Query_Sum(now<<1|1,mid+1,r,ql,qr);
return s;
}
int Query_Max(int now,int l,int r,int ql,int qr)
{
if (l>=ql&&r<=qr) return mx[now];
if (rev[now]) pushdown(now);
int mid=l+r>>1,s=-1e9;
if (ql<=mid) s=max(s,Query_Max(now<<1,l,mid,ql,qr));
if (qr>mid) s=max(s,Query_Max(now<<1|1,mid+1,r,ql,qr));
return s;
}
int Query_Min(int now,int l,int r,int ql,int qr)
{
if (l>=ql&&r<=qr) return mn[now];
if (rev[now]) pushdown(now);
int mid=l+r>>1,s=1e9;
if (ql<=mid) s=min(s,Query_Min(now<<1,l,mid,ql,qr));
if (qr>mid) s=min(s,Query_Min(now<<1|1,mid+1,r,ql,qr));
return s;
}
int main()
{
n=gi();
for (int i=1;i<n;i++)
{
int u=gi()+1,v=gi()+1,w=gi();
a[++cnt]=(edge){v,head[u],w};head[u]=cnt;
a[++cnt]=(edge){u,head[v],w};head[v]=cnt;
}
dfs1(1,0);cnt=0;dfs2(1,1);
for (int e=1;e<n;e++)
Modify(1,1,n,dfn[ref[e]],a[e<<1].w);
m=gi();
while (m--)
{
scanf("%s",s);
if (s[0]=='C')
{
int i=gi(),w=gi();
Modify(1,1,n,dfn[ref[i]],w);
}
if (s[0]=='N')
{
int u=gi()+1,v=gi()+1;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
Reverse(1,1,n,dfn[top[u]],dfn[u]);
u=fa[top[u]];
}
if (dep[u]>dep[v]) swap(u,v);
if (u^v) Reverse(1,1,n,dfn[son[u]],dfn[v]);
}
if (s[0]=='S')
{
int u=gi()+1,v=gi()+1,ans=0;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
ans+=Query_Sum(1,1,n,dfn[top[u]],dfn[u]);
u=fa[top[u]];
}
if (dep[u]>dep[v]) swap(u,v);
if (u^v) ans+=Query_Sum(1,1,n,dfn[son[u]],dfn[v]);
printf("%d\n",ans);
}
if (s[0]=='M'&&s[1]=='A')
{
int u=gi()+1,v=gi()+1,ans=-1e9;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
ans=max(ans,Query_Max(1,1,n,dfn[top[u]],dfn[u]));
u=fa[top[u]];
}
if (dep[u]>dep[v]) swap(u,v);
if (u^v) ans=max(ans,Query_Max(1,1,n,dfn[son[u]],dfn[v]));
printf("%d\n",ans);
}
if (s[0]=='M'&&s[1]=='I')
{
int u=gi()+1,v=gi()+1,ans=1e9;
while (top[u]^top[v])
{
if (dep[top[u]]<dep[top[v]]) swap(u,v);
ans=min(ans,Query_Min(1,1,n,dfn[top[u]],dfn[u]));
u=fa[top[u]];
}
if (dep[u]>dep[v]) swap(u,v);
if (u^v) ans=min(ans,Query_Min(1,1,n,dfn[son[u]],dfn[v]));
printf("%d\n",ans);
}
}
return 0;
}