hdu 3966 Aragorn's Story
树链剖分模板题。更新一条链的值以及求一个点的值。
可以用线段树或树状数组来区间更新加单点查询。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include <bits/stdc++.h> #define Lson l,mid,rt<<1 #define Rson mid+1,r,rt<<1|1 using namespace std; const int M = 5e4+7; int head[M],sz[M],son[M],f[M],id[M],rnk[M],dep[M],top[M]; int tot,cnt,n,m,q,a[M],pos,ans; struct edge{ int v,next; }e[M<<1]; struct tree { int sum,lazy,l,r; }tree[M<<2]; void init(){ tot=cnt=0;memset(head,-1,sizeof(head)); } void add(int u,int v){ e[++cnt].v=v;e[cnt].next=head[u]; head[u]=cnt; } void dfs(int u,int fa,int d){ sz[u]=1,son[u]=-1,f[u]=fa,dep[u]=d; for(int i=head[u];~i;i=e[i].next){ int v=e[i].v; if(v==fa) continue; dfs(v,u,d+1); sz[u]+=sz[v]; if(son[u]==-1||sz[v]>sz[son[u]]) son[u]=v; } return ; } void dfs1(int u,int t){ id[u]=++tot; rnk[tot]=u; top[u]=t; if(son[u]==-1) return ; dfs1(son[u],t); for(int i=head[u];~i;i=e[i].next){ int v=e[i].v; if(v==f[u]||v==son[u]) continue; dfs1(v,v); } return ; } void Pushup(int rt){ tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum; } void Pushdown(int rt){ if(tree[rt].lazy){ int v=tree[rt].lazy; tree[rt].lazy=0; tree[rt<<1].lazy+=v; tree[rt<<1|1].lazy+=v; tree[rt<<1].sum+=v*(tree[rt<<1].r-tree[rt<<1].l+1); tree[rt<<1|1].sum+=v*(tree[rt<<1|1].r-tree[rt<<1|1].l+1); } } void build(int l,int r,int rt){ tree[rt].l=l;tree[rt].r=r;tree[rt].lazy=0; if(l==r){ tree[rt].sum=a[rnk[l]]; return ; } int mid=(l+r)>>1; build(Lson); build(Rson); Pushup(rt); } void update(int L,int R,int l,int r,int rt,int v){ if(L<=l&&r<=R){ tree[rt].lazy+=v; tree[rt].sum+=v*(r-l+1); return ; } Pushdown(rt); int mid=(l+r)>>1; if(L<=mid) update(L,R,Lson,v); if(R>mid) update(L,R,Rson,v); Pushup(rt); } void query(int l,int r,int rt){ if(l==r){ ans=tree[rt].sum; return ; } Pushdown(rt); int mid=(l+r)>>1; if(pos<=mid) query(Lson); if(pos>mid) query(Rson); return ; } void updates(int x,int y,int v){ int fx=top[x],fy=top[y]; while(fx!=fy){ if(dep[fx]>dep[fy]){ update(id[fx],id[x],1,n,1,v); x=f[fx],fx=top[x]; } else{ update(id[fy],id[y],1,n,1,v); y=f[fy],fy=top[y]; } } if(dep[x]<dep[y]) update(id[x],id[y],1,n,1,v); else update(id[y],id[x],1,n,1,v); } int main(){ while(~scanf("%d%d%d",&n,&m,&q)){ init(); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=m;i++){ int u,v; scanf("%d%d",&u,&v); add(u,v);add(v,u); } dfs(1,-1,1); dfs1(1,1); build(1,n,1); while(q--){ char c[2]; int c1,c2,k; scanf("%s",c); if(c[0]=='I'||c[0]=='D'){ scanf("%d%d%d",&c1,&c2,&k); if(c[0]=='D') k*=-1; updates(c1,c2,k); } else{ scanf("%d",&c1);pos=id[c1];query(1,n,1); printf("%d\n",ans); } } } return 0; }