• BZOJ4034 [HAOI2015]树上操作 树链剖分


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

    去博客园看该题解


    题目传送门 - BZOJ4034


    题意概括

    有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个

    操作,分为三种:
    操作 1 :把某个节点 x 的点权增加 a 。
    操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
    操作 3 :询问某个节点 x 到根的路径中所有点的点权和。

    题解

      树链剖分。

      然后对于子树修改,我们可以考虑dfs序

      树链剖分也是一种dfs序。

      单点修改更简单,对于懒惰的我来说,这就是区间修改(少写了一个void 2333)

      询问几乎是基础的树剖了,沿着树链往上走就可以了。


    代码

    #include <cstring>
    #include <cstdio>
    #include <algorithm>
    #include <cmath>
    #include <cstdlib>
    using namespace std;
    typedef long long LL;
    const int N=100005;
    struct Gragh{
    	int cnt,y[N*2],nxt[N*2],fst[N];
    	void clear(){
    		cnt=0;
    		memset(fst,0,sizeof fst);
    	}
    	void add(int a,int b){
    		y[++cnt]=b,nxt[cnt]=fst[a],fst[a]=cnt;
    	}
    }g;
    int n,m,cnp=0;
    int fa[N],son[N],size[N],depth[N],top[N],p[N],ap[N],outp[N];
    LL v[N],t[N*4],add[N*4],w[N*4];
    void Get_Gen_Info(int rt,int pre,int d){
    	size[rt]=1,fa[rt]=pre,depth[rt]=d,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);
    			size[rt]+=size[s];
    			if (son[rt]==-1||size[s]>size[son[rt]])
    				son[rt]=s;
    		}
    }
    void Get_Top(int rt,int tp){
    	top[rt]=tp;
    	ap[p[rt]=++cnp]=rt;
    	if (son[rt]==-1){
    		outp[rt]=cnp;
    		return;
    	}
    	Get_Top(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_Top(s,s);
    	}
    	outp[rt]=cnp;
    }
    void pushup(int rt){
    	int ls=rt<<1,rs=ls|1;
    	t[rt]=t[ls]+t[rs];
    }
    void build(int rt,int le,int ri){
    	w[rt]=ri-le+1;
    	add[rt]=0;
    	if (le==ri){
    		t[rt]=v[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 pushdown(int rt){
    	int ls=rt<<1,rs=ls|1;
    	LL &a=add[rt];
    	if (a){
    		t[ls]+=w[ls]*a,add[ls]+=a;
    		t[rs]+=w[rs]*a,add[rs]+=a;
    		a=0;
    	}
    }
    void update(int rt,int le,int ri,int xle,int xri,LL d){
    	if (le>xri||ri<xle)
    		return;
    	if (xle<=le&&ri<=xri){
    		t[rt]+=w[rt]*d,add[rt]+=d;
    		return;
    	}
    	pushdown(rt);
    	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
    	update(ls,le,mid,xle,xri,d);
    	update(rs,mid+1,ri,xle,xri,d);
    	pushup(rt);
    }
    LL query(int rt,int le,int ri,int xle,int xri){
    	if (le>xri||ri<xle)
    		return 0;
    	if (xle<=le&&ri<=xri)
    		return t[rt];
    	pushdown(rt);
    	int mid=(le+ri)>>1,ls=rt<<1,rs=ls|1;
    	return query(ls,le,mid,xle,xri)+query(rs,mid+1,ri,xle,xri);
    }
    LL Tquery(int a){
    	int f=top[a];
    	LL ans=0;
    	while (a){
    		ans+=query(1,1,n,p[f],p[a]);
    		a=fa[f],f=top[a];
    	}
    	return ans;
    }
    int main(){
    	scanf("%d%d",&n,&m);
    	for (int i=1;i<=n;i++)
    		scanf("%lld",&v[i]);
    	g.clear();
    	for (int i=1,a,b;i<n;i++){
    		scanf("%d%d",&a,&b);
    		g.add(a,b);
    		g.add(b,a);
    	}
    	Get_Gen_Info(1,0,0);
    	Get_Top(1,1);
    	build(1,1,n);
    	for (int i=1;i<=m;i++){
    		int op,a,b;
    		scanf("%d",&op);
    		if (op==1){
    			scanf("%d%d",&a,&b);
    			update(1,1,n,p[a],p[a],b);
    		}
    		if (op==2){
    			scanf("%d%d",&a,&b);
    			update(1,1,n,p[a],outp[a],b);
    		}
    		if (op==3){
    			scanf("%d",&a);
    			printf("%lld
    ",Tquery(a));
    		}
    	}
    	return 0;
    }
    

      

  • 相关阅读:
    小程序入坑记:TodoList
    小程序入坑记:懒加载
    JS案例:Socket聊天室(两种方式)
    JS案例:继承和深复制
    React(四)进阶
    React(三)TableBar,ToDoList,Redux案例
    【Junit_Ant】使用Eclipse自带的Junit和Ant,生成测试报告
    【Loadrunner_WebService接口】对项目中的GetProduct接口生成性能脚本
    【Jmeter_WebService接口】对项目中的GetProduct接口生成性能脚本
    【Loadrunner_Http接口】使用Loadrunner对天气信息的接口编写脚本
  • 原文地址:https://www.cnblogs.com/zhouzhendong/p/BZOJ4034.html
Copyright © 2020-2023  润新知