记录编号	351360	评测结果	AAAAAAAAAA
题目名称	新型武器	最终得分	100
用户昵称	Smile	是否通过	通过
代码语言	C++	运行时间	4.487 s
提交时间	2016-11-16 14:54:20	内存使用	33.50 MiB

#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;
}