BZOJ 2836 魔法树 树链剖分

Description

Input

Output

Sample Input

4
0 1
1 2
2 3
4
Add 1 3 1
Query 0
Query 1
Query 2

Sample Output

3
3
2

HINT

传送门

今天考试考了去年day 1的题……只会100+40+60的暴力。

渣死了，心情不好。

看到有一题树剖裸题……那就随手写了。。。

数组要大点，线段树维护有毒。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int 
	N=150005;
int n,Ecnt,cnt;
struct Edge{
	int next,to;
}E[N<<1];int head[N];
struct Tree{
	int pre,top,tid,deep,sz,son,maxid;
}tree[N];
struct Segment{
	ll mark,sum;
}seg[N<<2];
void add(int u,int v){
	E[++Ecnt].next=head[u];
	E[Ecnt].to=v;
	head[u]=Ecnt;
}
void up(int u){
	seg[u].sum=seg[u<<1].sum+seg[u<<1|1].sum;
}
void down(int u,int L,int R){
	if (u && seg[u].mark){
		int l=u<<1,r=u<<1|1;
		int mid=(L+R)>>1;
		seg[l].mark+=seg[u].mark;
		seg[r].mark+=seg[u].mark;
		seg[l].sum+=seg[u].mark*(ll)(mid-L+1);
		seg[r].sum+=seg[u].mark*(ll)(R-mid);
		seg[u].mark=0LL;
	}
}
void dfs1(int u,int pre,int depth){
	tree[u].pre=pre;
	tree[u].sz=1;
	tree[u].deep=depth;
	tree[u].son=0;
	int maxs=0;
	for (int i=head[u];i;i=E[i].next){
		int j=E[i].to;
		if (j==pre) continue;
		dfs1(j,u,depth+1);
		tree[u].sz+=tree[j].sz;
		if (maxs<tree[j].sz)
			maxs=tree[j].sz,tree[u].son=j;
	}
}
void dfs2(int u,int ancestor){
	tree[u].maxid=tree[u].tid=++cnt;
	tree[u].top=ancestor;
	if (tree[u].son) dfs2(tree[u].son,ancestor);
	tree[u].maxid=max(tree[u].maxid,tree[tree[u].son].maxid);
	for (int i=head[u];i;i=E[i].next){
		int j=E[i].to;
		if (j==tree[u].pre || j==tree[u].son) continue;
		dfs2(j,j);
		tree[u].maxid=max(tree[u].maxid,tree[j].maxid);
	}
}
void update(int id,int l,int r,int gl,int gr,ll num){
	down(id,l,r);
	if (l>=gl && r<=gr){
		seg[id].mark+=num;
		seg[id].sum+=(ll)(r-l+1)*num;
		return;
	}
	int mid=(l+r)>>1;
	if (gl<=mid) update(id<<1,l,mid,gl,gr,num);
	if (gr>mid) update(id<<1|1,mid+1,r,gl,gr,num);
	up(id);
}
ll query(int id,int l,int r,int gl,int gr){
	down(id,l,r);
	if (l>=gl && r<=gr) return seg[id].sum;
	int mid=(l+r)>>1;
	ll t=0LL;
	if (gl<=mid) t+=query(id<<1,l,mid,gl,gr);
	if (gr>mid) t+=query(id<<1|1,mid+1,r,gl,gr);
	return t;
}
void change(int u,int v,ll d){
	while (tree[u].top!=tree[v].top){
		if (tree[tree[u].top].deep<tree[tree[v].top].deep) swap(u,v);
		update(1,1,n,tree[tree[u].top].tid,tree[u].tid,d);
		u=tree[tree[u].top].pre;
	}
	if (tree[u].deep<tree[v].deep) swap(u,v);
	update(1,1,n,tree[v].tid,tree[u].tid,d);
}
ll querysum(int u){
	return query(1,1,n,tree[u].tid,tree[u].maxid);
}
int main(){
	cin>>n;
	int x,y;Ecnt=0;
	for (int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		x++,y++;
		add(x,y),add(y,x);
	}
	dfs1(1,-1,0);
	cnt=0,dfs2(1,1);
	int m;cin>>m;
	ll d;char s[10];
	while (m--){
		scanf("%s",s);
		if (s[0]=='A'){
			scanf("%d%d%lld",&x,&y,&d);
			x++,y++;
			change(x,y,d);
		} else{
			scanf("%d",&x);x++;
			printf("%lld\n",querysum(x));
		}
	}
	return 0;
}