记录编号 461827 评测结果 AAAAAAAAAA
题目名称 新型武器 最终得分 100
用户昵称 GravatarHzoi_Mafia 是否通过 通过
代码语言 C++ 运行时间 1.417 s
提交时间 2017-10-20 19:55:44 内存使用 76.99 MiB
显示代码纯文本
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
inline int read(){
	int sum(0);char ch(getchar());
	for(;ch<'0'||ch>'9';ch=getchar());
	for(;ch>='0'&&ch<='9';sum=sum*10+(ch^48),ch=getchar());
	return sum;
}
struct edge{
	int e;edge *n;
}*pre[300005];
inline void insert(int s,int e){
	edge *tmp(new edge);tmp->e=e;tmp->n=pre[s];pre[s]=tmp;
}
int n,q;
int w[300005];
int cnt,l[300005],r[300005],pos[300005],dep[300005];
inline void dfs(int u,int fa){
	l[u]=++cnt;pos[cnt]=u;
	for(edge *i=pre[u];i;i=i->n){
		int e(i->e);
		if(e==fa)continue;
		dep[e]=dep[u]+1;
		dfs(e,u);
	}
	r[u]=cnt;
}
int root[300005],sum[6000005],lch[6000005],rch[6000005];
inline void pushup(int x){
	sum[x]=sum[lch[x]]+sum[rch[x]];
}
inline void update(int &x,int pos,int w,int l,int r){
	if(!x)x=++cnt;
	if(l==r){sum[x]=w;return;}
	int mid((l+r)>>1);
	if(pos<=mid)update(lch[x],pos,w,l,mid);
	else update(rch[x],pos,w,mid+1,r);
	pushup(x);
}
inline int query(int x,int ll,int rr,int l,int r){
	if(ll==l&&r==rr)return sum[x];
	int mid((l+r)>>1);
	if(rr<=mid)return query(lch[x],ll,rr,l,mid);
	if(mid<ll)return query(rch[x],ll,rr,mid+1,r);
	return query(lch[x],ll,mid,l,mid)+query(rch[x],mid+1,rr,mid+1,r);
}
int main(){
	freopen("newweapon.in","r",stdin);
	freopen("newweapon.out","w",stdout);
	n=read();
	for(int i=1;i<=n;++i)w[i]=read();
	for(int i=1;i<n;++i){
		int x(read()),y(read());
		insert(x,y);insert(y,x);
	}
	dfs(1,0);
	for(int i=1;i<=n;++i)update(root[dep[i]],l[i],w[i],1,n);
	q=read();
	while(q--){
		int op(read()),x(read()),y(read());
		if(op==1)update(root[dep[x]],l[x],y,1,n);
		else printf("%d\n",query(root[dep[x]+y],l[x],r[x],1,n));
	}
}