WA 50pts 求调
查看原帖
WA 50pts 求调
536439
YONIC楼主2022/3/19 08:58
#include<bits/stdc++.h>
#define int long long
#define MAXN (int)(1e5+5)
using namespace std;
int head[MAXN<<1],tot,seg[MAXN<<2],lazy[MAXN<<2],son[MAXN],fa[MAXN],d[MAXN],size[MAXN],top[MAXN],id[MAXN],w[MAXN],w2[MAXN];
int n,m,num,u,v,flag,x,y,z;
struct edge{int next,to;}h[MAXN<<1];
void add(int u,int v){
	h[++tot].next=head[u];
	head[u]=tot;
	h[tot].to=v;
}
void addlazy(int x,int l,int r,int d){
	lazy[x]+=d;
	seg[x]+=d*(r-l+1);
}
void pushdown(int x,int l,int r){
	if(lazy[x]){
		int mid=(l+r)>>1;
		addlazy(x<<1,l,mid,lazy[x]);
		addlazy(x<<1|1,mid+1,r,lazy[x]);
		lazy[x]=0;
	}
}
void build(int x,int l,int r){
	lazy[x]=0;
	if(l==r){
		seg[x]=w2[l];
		return;
	}
	int mid=(l+r)>>1;
	build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
	seg[x]=seg[x<<1]+seg[x<<1|1];
}
void update_seg(int nl,int nr,int x,int l,int r,int d){
	if(l>=nl&&r<=nr){
		addlazy(x,l,r,d);
		return;
	}
	pushdown(x,l,r);
	int mid=(l+r)>>1;
	if(nl<=mid) update_seg(nl,nr,x<<1,l,mid,d);
	if(nr>mid) update_seg(nl,nr,x<<1|1,mid+1,r,d);
	seg[x]=seg[x<<1]+seg[x<<1|1];
}
int ask_seg(int nl,int nr,int x,int l,int r){
	if(l>=nl&&r<=nr) return seg[x];
	pushdown(x,l,r);
	int res=0;
	int mid=(l+r)>>1;
	if(nl<=mid) res+=ask_seg(nl,nr,x<<1,l,mid);
	if(nr>mid) res+=ask_seg(nl,nr,x<<1|1,mid+1,r);
	return res;
}
void DFS1(int x,int father){
	d[x]=d[father]+1;
	fa[x]=father;
	size[x]=1;
	for(int i=head[x];~i;i=h[i].next){
		int y=h[i].to;
		if(y!=father){
			fa[y]=x;
			DFS1(y,x);
			size[x]+=size[y];
			if(!son[x]||size[son[x]]<size[y]) son[x]=y;
		}
	}
}
void DFS2(int x,int topx){
	id[x]=++num;
	w2[num]=w[x];
	top[x]=topx;
	if(!son[x]) return;
	DFS2(son[x],topx);
	for(int i=head[x];~i;i=h[i].next){
		int y=h[i].to;
		if(y!=fa[x]&&y!=son[x]) DFS2(y,y);
	}
}
void update(int x,int y,int z){
	while(top[x]!=top[y]){
		if(d[top[x]]<d[top[y]]) swap(x,y);
		update_seg(id[top[x]],id[x],1,1,n,z);
		x=fa[top[x]];
	}
	if(d[x]>d[y]) swap(x,y);
	update_seg(id[x],id[y],1,1,n,z);
}
int ask(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		if(d[top[x]]<d[top[y]]) swap(x,y);
		ans+=ask_seg(id[top[x]],id[x],1,1,n);
		x=fa[top[x]];
	}
	if(d[x]>d[y])swap(x,y);
	ans+=ask_seg(id[x],id[y],1,1,n);
	return ans;
}
signed main(){
	for(int i=1;i<=MAXN;++i){
		h[i].next=-1;
		head[i]=-1;
	}
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) scanf("%d",w+i);
	for(int i=1;i<n;++i){
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	DFS1(1,0);
	DFS2(1,1);
	build(1,1,n);
	while(m--){
		scanf("%d",&flag);
		if(flag==1){
			scanf("%d%d",&x,&y);
			update(x,x,y);
		}
		if(flag==2){
			scanf("%d%d",&x,&y);
			update_seg(id[x],id[x]+size[x]-1,1,1,n,y);
		}
		if(flag==3){
			scanf("%d",&x);
			printf("%d\n",ask(1,x));
		}
	}
	return 0;
}
2022/3/19 08:58
加载中...