分析:
树链剖分练手题
注意线段树是建立在dfs序上的,所以子树一定在线段树上是一个连续的区间
子树加的时候直接线段树区间加即可
询问树链信息的时候,因为top,deep,pre数组都是建立在原树上的
所以传入的是原树上的结点编号(输入编号)
这道题也算是一个史前巨坑了,以前一直A不了
今天终于一A了
tip
开ll
//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
using namespace std;
const int N=100010;
int pre[N],top[N],son[N],size[N],out[N],num[N],shu[N],clo=0,deep[N];
int n,m;
ll a[N];
struct Tree{
int x,y;
ll sum,lazy;
};
Tree t[N<<2];
struct node{
int x,y,nxt;
};
node way[N<<1];
int st[N],tot=0;
void add(int u,int w)
{
tot++;
way[tot].x=u;way[tot].y=w;way[tot].nxt=st[u];st[u]=tot;
tot++;
way[tot].x=w;way[tot].y=u;way[tot].nxt=st[w];st[w]=tot;
}
void dfs_1(int now,int fa,int dep)
{
deep[now]=dep;
pre[now]=fa;
size[now]=1;
int maxx=0;
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa)
{
dfs_1(way[i].y,now,dep+1);
size[now]+=size[way[i].y];
if (size[way[i].y]>maxx)
{
maxx=size[way[i].y];
son[now]=way[i].y;
}
}
}
void dfs_2(int now,int fa)
{
if (son[fa]!=now) top[now]=now;
else top[now]=top[fa];
num[now]=++clo; shu[num[now]]=now;
if (son[now])
{
dfs_2(son[now],now);
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa&&way[i].y!=son[now])
dfs_2(way[i].y,now);
}
out[now]=clo;
}
void update(int bh)
{
t[bh].sum=t[bh<<1].sum+t[(bh<<1)+1].sum;
}
void push(int bh)
{
int lc=bh<<1;
int rc=(bh<<1)+1;
if (t[bh].lazy&&t[bh].x!=t[bh].y)
{
t[lc].sum+=(t[lc].y-t[lc].x+1)*t[bh].lazy;
t[lc].lazy+=t[bh].lazy;
t[rc].sum+=(t[rc].y-t[rc].x+1)*t[bh].lazy;
t[rc].lazy+=t[bh].lazy;
t[bh].lazy=0;
}
}
void build(int bh,int l,int r)
{
t[bh].x=l; t[bh].y=r; t[bh].lazy=0;
if (l==r)
{
t[bh].sum=a[shu[l]];
return;
}
int mid=(l+r)>>1;
build(bh<<1,l,mid);
build((bh<<1)+1,mid+1,r);
update(bh);
}
void add(int bh,int l,int r,ll z)
{
push(bh);
if (t[bh].x>=l&&t[bh].y<=r)
{
t[bh].sum+=(t[bh].y-t[bh].x+1)*z;
t[bh].lazy+=z;
return;
}
int mid=(t[bh].x+t[bh].y)>>1;
if (l<=mid) add(bh<<1,l,r,z);
if (r>mid) add((bh<<1)+1,l,r,z);
update(bh);
}
ll ask(int bh,int l,int r)
{
push(bh);
if (t[bh].x>=l&&t[bh].y<=r)
{
return t[bh].sum;
}
int mid=(t[bh].x+t[bh].y)>>1;
ll ans=0;
if (l<=mid) ans+=ask(bh<<1,l,r);
if (r>mid) ans+=ask((bh<<1)+1,l,r);
return ans;
}
ll asksum(int u,int w)
{
ll ans=0;
int f1=top[u];
int f2=top[w];
while (f1!=f2)
{
if (deep[f1]<deep[f2]) swap(u,w),swap(f1,f2); //
ans+=ask(1,num[f1],num[u]);
u=pre[f1];
f1=top[u];
}
if (num[u]>num[w]) swap(u,w);
ans+=ask(1,num[u],num[w]);
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%lld",&a[i]);
for (int i=1;i<n;i++)
{
int u,w;
scanf("%d%d",&u,&w);
add(u,w);
}
dfs_1(1,0,1);
dfs_2(1,0);
build(1,1,n);
for (int i=1;i<=m;i++)
{
int opt,x,y;
scanf("%d",&opt);
if (opt==1)
{
scanf("%d%d",&x,&y);
add(1,num[x],num[x],(ll)y);
}
else if (opt==2)
{
scanf("%d%d",&x,&y);
add(1,num[x],out[x],(ll)y);
}
else
{
scanf("%d",&x);
printf("%lld\n",asksum(1,x));
}
}
return 0;
}