比赛 |
20161116 |
评测结果 |
AAAATTTTTT |
题目名称 |
新型武器 |
最终得分 |
40 |
用户昵称 |
cwm大佬%%% |
运行时间 |
6.012 s |
代码语言 |
C++ |
内存使用 |
7.47 MiB |
提交时间 |
2016-11-16 11:10:15 |
显示代码纯文本
#include<cstdio>
#include<cstring>
const int N=300000+10;
struct P{
int to,nex;
};
P tree[N*2];
int head[N];
int top=0;
int v[N],fa[N];
int n;
void init_fa(int p=1,int f=1){
fa[p]=f;
for(int i=head[p];i!=-1;i=tree[i].nex)if(tree[i].to!=f)init_fa(tree[i].to,p);
}
int dfs(int p,int rest){
if(p==-1)return 0;
if(rest==0)return v[p];
int ret=0;
for(int i=head[p];i!=-1;i=tree[i].nex)if(tree[i].to!=fa[p])
ret+=dfs(tree[i].to,rest-1);
return ret;
}
void add_edge(int fro,int to){
tree[top]=(P){to,head[fro]};
head[fro]=top++;
tree[top]=(P){fro,head[to]};
head[to]=top++;
}
int main()
{
freopen("newweapon.in","r",stdin);
freopen("newweapon.out","w",stdout);
memset(head,-1,sizeof(head));
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&v[i]);
for(int i=1;i<n;i++){
int fro,to;
scanf("%d%d",&fro,&to);
add_edge(fro,to);
}
init_fa();
//for(int i=1;i<=n;i++)printf("%d ",fa[i]);putchar('\n');
int q;
scanf("%d",&q);
for(int i=0;i<q;i++){
int type,u,a;
scanf("%d%d%d",&type,&u,&a);
if(type==1)v[u]=a;
else printf("%d\n",dfs(u,a));
}
return 0;
}