zoukankan      html  css  js  c++  java
  • 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
    ",find(a,b));
    		}
    	}
    	return 0;
    }
    

      

  • 相关阅读:
    2.16.8.内核启动的C语言阶段5
    2.16.7.内核启动的C语言阶段4
    2.16.6.内核启动的C语言阶段3
    2.16.5.内核启动的C语言阶段2
    JAVA_SE基础——34.static修饰成员变量
    JAVA_SE基础——33.this关键字的练习
    JAVA_SE基础——32.this关键字调用本类的构造方法
    JAVA_SE基础——31.this关键字
    JAVA类的方法调用和变量(全套)
    JAVA_SE基础——30.构造代码块
  • 原文地址:https://www.cnblogs.com/zhouzhendong/p/POJ3237.html
Copyright © 2011-2022 走看看