【树链剖分】树Tree(POJ3237)

                                          【题目背景】

给你一颗具有n个结点的树,结点编号为1到n且边的编号为1到n-1。每条边有一个边权。现要求模拟下列三种操作:
1.CHANGE i v:修改第i条边的权值为v
2.NEGATE a b:把a点到b点之间路径的边权全部取反;
3.QUERY a b:询问a点到b点之间路径的最大边权;

【输入格式】

第一行为一个整数N(N<=30000);
接下来N-1行,每行包含三个整数a,b,c,表示a到b有一条边,边权为c。边的编号是按照输入的顺序。接下来是若干个询问,以一行一个单词“DONE”作为这组数据的结束。

【输出格式】

对于每个“QUERY”输出一行。


边权树链剖分,在维护最大值的同时维护最小值,取反时最大值等于最小值的相反,最小值等于最大值的相反

在标记覆盖时,注意不要直接下传标记,因为可能本身自带标记,此时就相当于取消标记

#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[30005];
struct Tree
{
	int maxx,minn;
}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;
	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;
	tag[k<<1|1]=tag[k]-tag[k<<1|1];
	tag[k]=0;
}
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;
		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);
	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);
}
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=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);
	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);
}
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);
	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);
}
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]);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		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];
	while(scanf("%s",s))
	{
		if(s[0]=='D') break;
		int x,y;
		scanf("%d%d",&x,&y);
		if(s[0]=='Q')
		{
			ask(x,y);
			printf("%d\n",ans);
		}
		else if(s[0]=='N')
		{
			uprange(x,y);
		}
		else
		{
			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);
		}
	}
}

猜你喜欢

转载自blog.csdn.net/Dy_Dream/article/details/82832509
今日推荐