树链剖分基础题
#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 60004
int val[maxn];
int fir[maxn],nex[maxn],v[maxn],e_max;
int son[maxn],fa[maxn],siz[maxn],deep[maxn],top[maxn],pos[maxn],tot;
long long c[maxn+10];
void init_()
{
memset(c,0,sizeof c);
memset(siz,0,sizeof siz);
memset(son,-1,sizeof son);
memset(fir,-1,sizeof fir);
tot=1;
e_max=0;
}
void add_edge(int s,int t)
{
int e=e_max++;
v[e]=t;
nex[e]=fir[s];
fir[s]=e;
}
int lowbit(int x)
{
return x&-x;
}
void update(int x,int val)
{
for(;x<=maxn;x+=lowbit(x)) c[x]+=val;
}
long long query(int x)
{
if(x==0) return 0;
long long sum=0;
for(;x>0;x-=lowbit(x)) sum+=c[x];
return sum;
}
void dfs1(int k,int pre,int d)
{
deep[k]=d;
siz[k]++;
fa[k]=pre;
for(int i=fir[k];~i;i=nex[i])
{
int e=v[i];
if(e!=pre)
{
dfs1(e,k,d+1);
siz[k]+=siz[e];
if(son[k]==-1||siz[son[k]]<siz[e]) son[k]=e;
}
}
}
void dfs2(int k,int sp)
{
pos[k]=tot++;
top[k]=sp;
update(pos[k],val[k]);
if(son[k]==-1) return ;
dfs2(son[k],sp);
for(int i=fir[k];~i;i=nex[i])
{
int e=v[i];
if(e!=son[k]&&e!=fa[k])
{
dfs2(e,e);
}
}
}
void Query(int s,int t)
{
long long ans=0;
int f1=top[s],f2=top[t];
while(f1!=f2)
{
if(deep[f1]<deep[f2]) swap(f1,f2),swap(s,t);
ans+=query(pos[s])-query(pos[f1]-1);
s=fa[f1];
f1=top[s];
}
if(deep[s]>deep[t]) swap(s,t);
ans+=query(pos[t])-query(pos[s]-1);
printf("%lld\n",ans);
}
int main()
{
int T;
int ca=1;
scanf("%d",&T);
while(T--)
{
init_();
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%d",&val[i]);
}
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
add_edge(a,b);
add_edge(b,a);
}
dfs1(1,-1,1);
dfs2(1,1);
int q;
scanf("%d",&q);
printf("Case %d:\n",ca++);
while(q--)
{
int op;
scanf("%d",&op);
if(op==0)
{
int a,b;
scanf("%d%d",&a,&b);
Query(a,b);
}
else
{
int i,ti;
scanf("%d%d",&i,&ti);
update(pos[i],-( query(pos[i])-query(pos[i]-1) ));
update(pos[i],ti);
}
}
}
return 0;
}