记录编号 307929 评测结果 AAAAAAAAAA
题目名称 [HAOI 2015]树上操作 最终得分 100
用户昵称 Gravatar521 是否通过 通过
代码语言 C++ 运行时间 1.828 s
提交时间 2016-09-16 15:51:24 内存使用 83.10 MiB
显示代码纯文本
#include<stdio.h>
#define LL long long
inline void read(int&x)
{
	char ch;bool flag=false;
	while(ch=getchar(),ch<'0'||ch>'9')if(ch=='-')flag=1;
	x=ch^'0';
	while(ch=getchar(),ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^'0');
	if(flag) x=-x;
}
const int Size=1000010;
int last[Size],son[Size],fa[Size],siz[Size],top[Size],dep[Size],tid[Size],rank[Size],tn;
LL sum[Size<<2],lazy[Size<<2],val[Size];
int to[Size<<1],next[Size<<1],head[Size],num;
inline void add(int u,int v)
{
	to[++num]=v,next[num]=head[u],head[u]=num;
}
void dfs1(int u,int f,int d)
{
	son[u]=0,fa[u]=f,siz[u]=1,dep[u]=d;
	for(int i=head[u];i;i=next[i]){
		int v=to[i];
		if(v==f) continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if(siz[v]>siz[son[u]]) son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	top[u]=tp,tid[u]=++tn,rank[tid[u]]=u;
	if(son[u]) dfs2(son[u],tp);
	for(int i=head[u];i;i=next[i])
		if(to[i]!=son[u]&&to[i]!=fa[u])
			dfs2(to[i],to[i]);
	last[u]=tn;
}
void build(int l,int r,int rt)
{
	if(l>r) return ;
	if(l==r){
		sum[rt]=val[rank[l]];
		return ;
	}
	int mid=l+r>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
inline void pushdown(int rt,int x)
{
	if(!lazy[rt]) return ;
	lazy[rt<<1]+=lazy[rt];
	lazy[rt<<1|1]+=lazy[rt];
	sum[rt<<1]+=lazy[rt]*(x-(x>>1));
	sum[rt<<1|1]+=lazy[rt]*(x>>1);
	lazy[rt]=0;
}
void update(int L,int R,LL x,int l,int r,int rt)
{
	if(L>R||l>r) return ;
	if(l>=L&&r<=R){
		sum[rt]+=x*(r-l+1);
		lazy[rt]+=x;
		return ;
	}
	pushdown(rt,r-l+1);
	int mid=l+r>>1;
	if(mid>=L) update(L,R,x,l,mid,rt<<1);
	if(mid<R) update(L,R,x,mid+1,r,rt<<1|1);
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
inline void swap(int &x,int &y){x^=y,y^=x,x^=y;}
LL query(int L,int R,int l,int r,int rt)
{
	if(L>R||l>r) return 0;
	if(l>=L&&r<=R) return sum[rt];
	pushdown(rt,r-l+1);
	int mid=l+r>>1;
	LL ret=0;
	if(mid>=L) ret+=query(L,R,l,mid,rt<<1);
	if(mid<R) ret+=query(L,R,mid+1,r,rt<<1|1);
	return ret;
}
LL find(int u,int v)
{
	LL ret=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		ret+=query(tid[top[u]],tid[u],1,tn,1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	ret+=query(tid[u],tid[v],1,tn,1);
	return ret;
}
int _521()
{
#define COGS
#ifdef COGS
	freopen("haoi2015_t2.in","r",stdin);
	freopen("haoi2015_t2.out","w",stdout);
#else
	freopen("1.in","r",stdin);
#endif
	int i,n,m;
	read(n),read(m);
	for(i=1;i<=n;i++) scanf("%lld",&val[i]);
	for(i=1;i<n;i++){
		int u,v;
		read(u),read(v);
		add(u,v),add(v,u);
	}
	dfs1(1,0,1),dfs2(1,1);
	build(1,tn,1);
	while(m--){
		int a,b;LL c;
		read(a);
		if(a==1){
			read(b),scanf("%lld",&c);
			update(tid[b],tid[b],c,1,tn,1);
		}else if(a==2){
			read(b),scanf("%lld",&c);
			update(tid[b],last[b],c,1,tn,1);
		}else{
			read(b);
			printf("%lld\n",find(1,b));
		}
	}
	return 0;
}
int _520=_521();
int main(){;}