【链接】
bzoj4034
【解题报告】
此题好像是树剖模板,但是好像线段树+DFS序也可以解决。
只需要在DFS序中,对于一个点的进栈与出栈,只需要标记进栈为正,出栈为负,就可以解决了。
#include<cstdio>
#include<cstring>
#define LL long long
using namespace std;
const int maxn=100005,maxm=200005,maxv=800005;
int n,m,tot,cnt,a[maxn],lnk[maxn],son[maxm],nxt[maxm],in[maxm],out[maxm],p[maxm];
bool vis[maxm];
struct Tree
{
LL s[maxv],tag[maxv],flg[maxv];
void Pushup(int d) {s[d]=s[d<<1]+s[d<<1|1];}
void Pushdown(int d)
{
if (!tag[d]) return;
s[d<<1]+=(LL)tag[d]*flg[d<<1]; s[d<<1|1]+=(LL)tag[d]*flg[d<<1|1];
tag[d<<1]+=tag[d]; tag[d<<1|1]+=tag[d]; tag[d]=0;
}
void Build(int d,int l,int r)
{
if (l==r) {s[d]=p[l]; tag[d]=0; if (vis[l]) flg[d]=1; else flg[d]=-1; return;}
int mid=(r-l>>1)+l;
Build(d<<1,l,mid); Build(d<<1|1,mid+1,r);
Pushup(d); flg[d]=flg[d<<1]+flg[d<<1|1];
}
void Insert(int d,int l,int r,int x,int y,int p)
{
if (l==x&&r==y) {s[d]+=p*flg[d]; tag[d]+=p; return;}
int mid=(r-l>>1)+l; Pushdown(d);
if (y<=mid) Insert(d<<1,l,mid,x,y,p);
else if (x>mid) Insert(d<<1|1,mid+1,r,x,y,p);
else Insert(d<<1,l,mid,x,mid,p),Insert(d<<1|1,mid+1,r,mid+1,y,p);
Pushup(d);
}
LL Query(int d,int l,int r,int x,int y)
{
if (l==x&&r==y) return s[d];
int mid=(r-l>>1)+l; Pushdown(d);
if (y<=mid) return Query(d<<1,l,mid,x,y);
else if (x>mid) return Query(d<<1|1,mid+1,r,x,y);
else return Query(d<<1,l,mid,x,mid)+Query(d<<1|1,mid+1,r,mid+1,y);
}
}tr;
inline int Read()
{
int res=0,f=1;
char ch=getchar(),cc=ch;
while (ch<'0'||ch>'9') cc=ch,ch=getchar();
if (cc=='-') f=-1;
while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=getchar();
return res*f;
}
void Add(int x,int y) {son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;}
void Dfs(int x,int fa)
{
p[in[x]=++cnt]=a[x]; vis[cnt]=1;
for (int j=lnk[x]; j; j=nxt[j])
if (son[j]!=fa) Dfs(son[j],x);
p[out[x]=++cnt]=-a[x]; vis[cnt]=0;
}
int main()
{
freopen("4034.in","r",stdin);
freopen("4034.out","w",stdout);
n=Read(); m=Read(); tot=0; cnt=0;
memset(lnk,0,sizeof(lnk));
for (int i=1; i<=n; i++) a[i]=Read();
for (int i=1,x,y; i<n; i++) x=Read(),y=Read(),Add(x,y),Add(y,x);
Dfs(1,0); tr.Build(1,1,2*n);
for (int i=1; i<=m; i++)
{
int z=Read(),x=Read(),p;
if (z==1) p=Read(),tr.Insert(1,1,2*n,in[x],in[x],p),tr.Insert(1,1,2*n,out[x],out[x],p);
else if (z==2) p=Read(),tr.Insert(1,1,2*n,in[x],out[x],p);
else printf("%lld\n",tr.Query(1,1,2*n,1,in[x]));
}
return 0;
}