树链剖分+线段树--bzoj2836: 魔法树

最新推荐文章于 2022-07-15 00:11:01 发布

原创最新推荐文章于 2022-07-15 00:11:01 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树链剖分 #线段树

线段树同时被 2 个专栏收录

14 篇文章

订阅专栏

树链剖分

10 篇文章

订阅专栏

本文深入探讨了树状数组与线段树在数据结构中的应用，特别是针对树剖分算法的优化策略。通过实例代码，详细解析了如何使用线段树进行区间更新和查询操作，以提高算法效率。同时，分享了作者在实现过程中的心得与教训，为读者提供了宝贵的实战经验。

树剖裸题···
然而太久没写树剖的我 $z z$ 到修改用 $u p d a t e$ 直接改
简直是 $m d z z$

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100005
#define LL long long
#define ls cur<<1
#define rs cur<<1|1
#define int LL
using namespace std;
int n,m,cnt,head[N],rt=1;
int tot,dfn[N],dep[N],son[N],siz[N],fa[N],top[N],rk[N];

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}

struct EDGE{
	int to,nxt;
}edge[N<<1];
inline void add(int x,int y){
	edge[++cnt].to=y; edge[cnt].nxt=head[x]; head[x]=cnt;
}

void dfs1(int u,int depth,int fat){
	dep[u]=depth,fa[u]=fat; siz[u]=1;
	int maxson=-1;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to; if(v==fat) continue;
		dfs1(v,depth+1,u); siz[u]+=siz[v];
		if(siz[v]>maxson) maxson=siz[v],son[u]=v;
	} return;
}

void dfs2(int u,int t){
	dfn[u]=++tot; rk[tot]=u; top[u]=t;
	if(!son[u]) return;
	dfs2(son[u],t);
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(!dfn[v]) dfs2(v,v);
	} return;
}

struct Node{
	int l,r; LL sum,laz;
}node[N<<2];

void build(int cur,int L,int R){
	if(L==R){
		node[cur].l=node[cur].r=L; node[cur].sum=node[cur].laz=0;
		return;
	}
	int mid=(L+R)>>1;
	build(ls,L,mid); build(rs,mid+1,R);
	node[cur].l=node[ls].l; node[cur].r=node[rs].r;
}

inline void pushdown(int cur){
	LL x=node[cur].laz; node[cur].laz=0;
	node[ls].sum+=x*(node[ls].r-node[ls].l+1);
	node[rs].sum+=x*(node[rs].r-node[rs].l+1);
	node[ls].laz+=x, node[rs].laz+=x;
}

inline void pushup(int cur){
	node[cur].sum=node[ls].sum+node[rs].sum;
}

void update(int cur,int L,int R,int c){
	if(L<=node[cur].l && node[cur].r<=R){
		node[cur].sum+=1LL*(node[cur].r-node[cur].l+1)*c;
		node[cur].laz+=c; return; 
	}
	pushdown(cur);
	int mid=(node[cur].l+node[cur].r)>>1;
	if(L<=mid) update(ls,L,R,c);
	if(mid<R) update(rs,L,R,c);
	pushup(cur); return;
}

LL query(int cur,int L,int R){
	if(L<=node[cur].l && node[cur].r<=R) return node[cur].sum;
	pushdown(cur);
	int mid=(node[cur].l+node[cur].r)>>1; LL res=0;
	if(L<=mid) res+=query(ls,L,R);
	if(mid<R) res+=query(rs,L,R);
	return res;
}

inline void change(int x,int y,LL z){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		update(1,dfn[top[x]],dfn[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update(1,dfn[x],dfn[y],z);
}

signed main(){
	n=rd();
	for(int i=1;i<n;i++){
		int x=rd(),y=rd(); x++,y++;
		add(x,y); add(y,x);
	}
	dfs1(rt,1,rt); dfs2(rt,rt);
	build(1,1,n);
	m=rd();
	while(m--){
		char s[10]; scanf("%s",s);
		if(s[0]=='A'){
			int x=rd(),y=rd(),z=rd(); x++,y++;
			change(x,y,z);
		}else{
			int x=rd(); x++;
			printf("%lld\n",query(1,dfn[x],dfn[x]+siz[x]-1));
		}
	}
	return 0;
}