bzoj 4034: [HAOI2015]树上操作 （树链剖分）

4034: [HAOI2015]树上操作

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 3553   Solved: 1118
[ Submit][ Status][ Discuss]

Description

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

操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

Input

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 

行每行三个正整数 fr, to ， 表示该树中存在一条边 (fr, to) 。再接下来 M 行，每行分别表示一次操作。其中

第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

Output

对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

Sample Input

5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

Sample Output

6
9
13

HINT

 对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。

Source

鸣谢bhiaibogf提供

[ Submit][ Status][ Discuss]

题解：树链剖分

#include
#include
#include
#include
#define N 200003
#define LL long long
using namespace std;
int tot,n,m,next[N],point[N],v[N],deep[N],fa[N],size[N],son[N];
LL tr[N*4],delta[N*4];
int val[N],l[N],r[N],q[N],pos[N],cnt,belong[N];
void add(int x,int  y)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y;
	tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int f)
{
	deep[x]=deep[f]+1;
	size[x]=1;
	for (int i=point[x];i;i=next[i])
	 if (v[i]!=f) {
	 	fa[v[i]]=x;
	 	dfs(v[i],x);
	 	size[x]+=size[v[i]];
	 	if (size[son[x]]mid) qjchange(now<<1|1,mid+1,r,ll,rr,v);
	update(now);
}
LL qjsum(int now,int l,int r,int ll,int rr)
{
	if (ll<=l&&r<=rr) return tr[now];
	int mid=(l+r)/2; LL ans=0;
	pushdown(now,l,r);
	if (ll<=mid) ans+=qjsum(now<<1,l,mid,ll,rr);
	if (rr>mid) ans+=qjsum(now<<1|1,mid+1,r,ll,rr);
	return ans;
}
LL solve(int x,int y)
{
	LL ans=0;
	while (belong[x]!=belong[y]) {
		if (deep[belong[x]]deep[y]) swap(x,y);
	ans+=qjsum(1,1,n,pos[x],pos[y]);
	return ans;
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) scanf("%d",&val[i]);
	for (int i=1;i