比赛 |
20161116 |
评测结果 |
AAAAAAAAAA |
题目名称 |
新型武器 |
最终得分 |
100 |
用户昵称 |
Smile |
运行时间 |
4.652 s |
代码语言 |
C++ |
内存使用 |
33.50 MiB |
提交时间 |
2016-11-16 11:57:46 |
显示代码纯文本
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=300000+10;
const int INF=0x3f3f3f3f;
const int LOG=20;
vector<int> G[maxn];
vector<int> ce[maxn];
int n, q, val[maxn];
int fa[maxn][LOG+1], dep[maxn];
void dfs(int pa, int u, int d) {
fa[u][0]=pa;
dep[u]=d;
ce[d].push_back(u);
for(int i=0; i<(int)G[u].size(); i++) {
if(G[u][i]!=pa) {
dfs(u, G[u][i], d+1);
}
}
}
void CSH() {
dfs(-1, 1, 0);
for(int k=0; k<LOG-1; k++) {
for(int i=1; i<=n; i++) {
if(fa[i][k]==-1) fa[i][k+1]=-1;
else fa[i][k+1]=fa[fa[i][k]][k];
}
}
}
bool LCA(int u, int v) {
if(dep[u]<dep[v]) swap(u, v);
for(int k=LOG-1; k>=0; k--) {
if(fa[u][k]!=-1 && dep[fa[u][k]]>=dep[v]) u=fa[u][k];
}
if(u==v) return 1;
else return 0;
}
int Qin() {
int x=0;
char c=getchar();
while(c<'0' || c>'9') c=getchar();
while(c>='0' && c<='9') x=x*10+c-'0', c=getchar();
return x;
}
int Query(int qd, int u) {
int ans=0;
for(int i=0; i<(int)ce[qd].size(); i++) {
int v=ce[qd][i];
if(LCA(v, u)) ans+=val[v];
}
return ans;
}
void init() {
n=Qin();
for(int i=1; i<=n; i++) val[i]=Qin();
for(int i=1; i<n; i++) {
int x=Qin(), y=Qin();
G[x].push_back(y);
G[y].push_back(x);
}
CSH();
q=Qin();
for(int i=1; i<=q; i++) {
int flag=Qin(), u=Qin(), v=Qin();
if(flag==1) {
val[u]=v;
}
else {
printf("%d\n", Query(dep[u]+v, u));
}
}
}
int main() {
freopen("newweapon.in", "r", stdin);
freopen("newweapon.out", "w", stdout);
init();
return 0;
}