POJ3237 Tree 树链剖分 线段树

欢迎访问~原文出处——博客园-zhouzhendong

去博客园看该题解

---

题目传送门 - POJ3237

---

题意概括

Description

给你由N个结点组成的树。树的节点被编号为1到N，边被编号为1到N-1。每一条边有一个权值。然后你要在树上执行一系列指令。指令可以是如下三种之一：
CHANGE i v：将第i条边的权值改成v。
NEGATE a b：将点a到点b路径上所有边的权值变成其相反数。
QUERY a b：找出点a到点b路径上各边的最大权值。

Input
多组数据，数据为T<=20，对于每组数据：
第一行有一个整数N（N<=10000）。
接下来N-1行每行有三个整数a,b,c，代表点a和点b之间有一条权值为c的边。这些边按照其编号从小到大给出。
接下来是若干条指令（不超过10^5条），都按照上面所说的格式。
最后一行是"DONE".

Output
对每个“QUERY”指令，输出一行，即路径上各边的最大权值。

Sample Input
1
3
1 2 1
2 3 2
QUERY 1 2
CHANGE 1 3
QUERY 1 2
DONE

Sample Output
1
3

---

题解

　　树链剖分+线段树单点修改、区间修改、区间询问。

　　对于取相反数的，我们只需要维护一个最大值，一个最小值，然后打一下懒标记就可以了。

　　在下传的时候，最大值反一反一定是最小的，最小值反一反一定是最大的，然后就好办了。

---

代码

#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int N=10005;
struct Edge{
	int cnt,y[N*2],z[N*2],nxt[N*2],fst[N];
	void clear(){
		cnt=0;
		memset(fst,0,sizeof fst);
	}
	void add(int a,int b,int c){
		y[++cnt]=b,z[cnt]=c,nxt[cnt]=fst[a],fst[a]=cnt;
	}
}g;
struct edge{
	int a,b,c;
}e[N];
int T,n,fa[N],size[N],fadis[N],depth[N],son[N],top[N],p[N],ap[N],cnp;
struct Tree{
	int ma,mi,add;
}t[N*4];
void Get_Gen_Info(int rt,int pre,int d){
	depth[rt]=d,fa[rt]=pre,size[rt]=1,son[rt]=-1;
	for (int i=g.fst[rt];i;i=g.nxt[i])
		if (g.y[i]!=pre){
			int s=g.y[i];
			Get_Gen_Info(s,rt,d+1);
			fadis[s]=g.z[i];
			size[rt]+=size[s];
			if (son[rt]==-1||size[s]>size[son[rt]])
				son[rt]=s;
		}
}
void Get_Pos(int rt,int tp){
	top[rt]=tp,p[rt]=++cnp,ap[cnp]=rt;
	if (son[rt]==-1)
		return;
	else
		Get_Pos(son[rt],tp);
	for (int i=g.fst[rt];i;i=g.nxt[i]){
		int s=g.y[i];
		if (s!=fa[rt]&&s!=son[rt])
			Get_Pos(s,s);
	}
}
void pushup(int rt){
	int ls=rt<<1,rs=ls|1;
	t[rt].mi=min(t[ls].mi,t[rs].mi);
	t[rt].ma=max(t[ls].ma,t[rs].ma);
}
void build(int rt,int le,int ri){
	t[rt].add=0;
	if (le==ri){
		t[rt].ma=t[rt].mi=fadis[ap[le]];
		return;
	}
	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
	build(ls,le,mid);
	build(rs,mid+1,ri);
	pushup(rt);
}
void fz(int &x,int &y){
	x=-x,y=-y;
	swap(x,y);
}
void pushdown(int rt){
	int &a=t[rt].add;
	int ls=rt<<1,rs=ls|1;
	if (a==0)
		return;
	t[ls].add^=1,t[rs].add^=1;
	fz(t[ls].ma,t[ls].mi);
	fz(t[rs].ma,t[rs].mi);
	a=0;
}
void change(int rt,int le,int ri,int pos,int v){
	if (le==ri){
		t[rt].mi=t[rt].ma=v;
		return;
	}
	pushdown(rt);
	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
	if (pos<=mid)
		change(ls,le,mid,pos,v);
	else
		change(rs,mid+1,ri,pos,v);
	pushup(rt);
}
void update(int rt,int le,int ri,int xle,int xri){
	if (le>xri||ri<xle)
		return;
	if (xle<=le&&ri<=xri){
		t[rt].add^=1;
		fz(t[rt].ma,t[rt].mi);
		return;
	}
	pushdown(rt);
	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
	update(ls,le,mid,xle,xri);
	update(rs,mid+1,ri,xle,xri);
	pushup(rt);
}
int query(int rt,int le,int ri,int xle,int xri){
	if (le>xri||ri<xle)
		return -1e9;
	if (xle<=le&&ri<=xri)
		return t[rt].ma;
	pushdown(rt);
	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
	return max(query(ls,le,mid,xle,xri),query(rs,mid+1,ri,xle,xri));
}
void fupdate(int a,int b){
	int f1=top[a],f2=top[b];
	while (f1!=f2){
		if (depth[f1]<depth[f2])
			swap(f1,f2),swap(a,b);
		update(1,1,n,p[f1],p[a]);
		a=fa[f1],f1=top[a];
	}
	if (a==b)
		return;
	if (depth[a]>depth[b])
		swap(a,b);
	update(1,1,n,p[son[a]],p[b]);
}
int find(int a,int b){
	int f1=top[a],f2=top[b],ans=-1e9;
	while (f1!=f2){
		if (depth[f1]<depth[f2])
			swap(f1,f2),swap(a,b);
		ans=max(ans,query(1,1,n,p[f1],p[a]));
		a=fa[f1],f1=top[a];
	}
	if (a==b)
		return ans;
	if (depth[a]>depth[b])
		swap(a,b);
	return max(ans,query(1,1,n,p[son[a]],p[b]));
}
int main(){
	scanf("%d",&T);
	while (T--){
		scanf("%d",&n);
		g.clear();
		for (int i=1,a,b,c;i<n;i++){
			scanf("%d%d%d",&a,&b,&c);
			g.add(a,b,c);
			g.add(b,a,c);
			e[i].a=a,e[i].b=b,e[i].c=c;
		}
		cnp=0,fadis[1]=0;
		Get_Gen_Info(1,0,0);
		Get_Pos(1,1);
		build(1,1,n);
		for (int i=1;i<n;i++)
			if (depth[e[i].a]>depth[e[i].b])
				swap(e[i].a,e[i].b);
		char str[10];
		int a,b;
		while (scanf("%s",str)&&str[0]!='D'){
			scanf("%d%d",&a,&b);
			if (str[0]=='C')
				change(1,1,n,p[e[a].b],b);
			else if (str[0]=='N')
				fupdate(a,b);
			else
				printf("%d\n",find(a,b));
		}
	}
	return 0;
}