比赛 |
20161116 |
评测结果 |
AAAATTTTTT |
题目名称 |
新型武器 |
最终得分 |
40 |
用户昵称 |
coolkid |
运行时间 |
6.021 s |
代码语言 |
C++ |
内存使用 |
10.32 MiB |
提交时间 |
2016-11-16 11:54:10 |
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int MAXN=300010;
struct Edge{
int to,nxt;
}edges[MAXN<<1];
int head[MAXN],cnt=0;
int val[MAXN];
vector<int>p[MAXN];
int dep[MAXN],n;
int fa[MAXN];
void addedge(int f,int t){
edges[cnt].to=t;
edges[cnt].nxt=head[f];
head[f]=cnt++;
}
void init(){
memset(fa,0,sizeof(fa));
memset(head,-1,sizeof(head));
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&val[i]);
for(int i=1;i<n;i++){
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);
addedge(y,x);
}
}
void dfs(int u){
for(int i=head[u];~i;i=edges[i].nxt){
int v=edges[i].to;
if(fa[u]==v) continue;
fa[v]=u;
dep[v]=dep[u]+1;
p[dep[v]].push_back(v);
dfs(v);
}
}
void work(){
memset(dep,0,sizeof(dep));
fa[1]=0;
dfs(1);
int Q;
scanf("%d",&Q);
while(Q--){
int opt;
scanf("%d",&opt);
if(opt==1){
int u,v;
scanf("%d%d",&u,&v);
val[u]=v;
}
else{
int u,c;
scanf("%d%d",&u,&c);
int dpt=dep[u]+c;
int ans=0;
for(int i=0;i<p[dpt].size();i++){
int v=p[dpt][i];
int F=v;
while(F!=0){
if(F==u){
ans+=val[v];
}
F=fa[F];
}
}
printf("%d\n",ans);
}
}
}
int main(){
freopen("newweapon.in","r",stdin);
freopen("newweapon.out","w",stdout);
init();
work();
return 0;
}