【树链剖分】BZOJ2157 旅游

【题目描述】

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);
		}
	}
}

猜你喜欢

转载自blog.csdn.net/Dy_Dream/article/details/83027961