最近在强化知识点深度,发现树链剖分不是很会写了。
回顾一下修改操作:
若两个点在同一条链上,则直接修改这段区间。
若不在同一条链上,修改深度较大的点到其链顶端的区间,同时将这个点变为他所在链顶端的父亲,循环操作直到这两个点在同一条链上,就可以用上一种方法了。
没有用LCA写是因为以前被坑过,不但没有这种方法好写,效率也不太让人满意。
主要是对第二种情况如何写有所遗忘,写道模版再给自己提个醒。
#include<iostream> #include<cstdio> #include<cstdlib> #include<string.h> #include<math.h> #include<algorithm> #include<vector> #include<queue> using namespace std; #define MAXN 50005 int n,m,q; vector<int> map[MAXN]; int size[MAXN],fa[MAXN],son[MAXN],val[MAXN],tid[MAXN],_tid[MAXN],dep[MAXN],top[MAXN]; int cnt; struct node { int l,r,val; }tree[MAXN<<2]; void dfs1(int s,int f,int d) { size[s]=1; fa[s]=f; dep[s]=d; int len=map[s].size(); for(int i=0;i<len;i++) { int e=map[s][i]; if(e==f)continue; dfs1(e,s,d+1); size[s]+=size[e]; if(son[s]==0) son[s]=e; else if(size[son[s]]<size[e]) son[s]=e; } } void dfs2(int s,int t) { tid[s]=++cnt; _tid[cnt]=s; top[s]=t; if(son[s]!=0) dfs2(son[s],t); int len=map[s].size(); for(int i=0;i<len;i++) { int e=map[s][i]; if(e!=fa[s] && e!=son[s]) dfs2(e,e); } } void build(int l,int r,int now) { tree[now].l=l; tree[now].r=r; tree[now].val=0; if(l==r) { tree[now].val=val[_tid[l]]; return ; } int mid=(l+r)>>1; build(l,mid,now<<1); build(mid+1,r,now<<1|1); } void down(int now) { tree[now<<1].val+=tree[now].val; tree[now<<1|1].val+=tree[now].val; tree[now].val=0; } void update(int l,int r,int now,int num) { if(l==tree[now].l && r==tree[now].r) { tree[now].val+=num; return ; } if(tree[now].val) down(now); int mid=(tree[now].l+tree[now].r)>>1; if(r<=mid) update(l,r,now<<1,num); else if(l>mid) update(l,r,now<<1|1,num); else { update(l,mid,now<<1,num); update(mid+1,r,now<<1|1,num); } } void change(int s,int e,int num) { while(top[s]!=top[e]) { if(dep[top[s]]<dep[top[e]]) swap(s,e); update(tid[top[s]],tid[s],1,num); s=fa[top[s]]; } if(dep[s]>dep[e]) swap(s,e); update(tid[s],tid[e],1,num); } int query(int l,int now) { if(tree[now].l==l && tree[now].r==l) return tree[now].val; if(tree[now].val) down(now); int mid=(tree[now].l+tree[now].r)>>1; if(l<=mid) return query(l,now<<1); else return query(l,now<<1|1); } int main() { while(cin>>n>>m>>q) { memset(size,0,sizeof(size)); memset(fa,0,sizeof(fa)); memset(tid,0,sizeof(tid)); memset(son,0,sizeof(son)); cnt=0; for(int i=1;i<=n;i++) { scanf("%d",&val[i]); map[i].clear(); } for(int i=1;i<n;i++) { int a,b; scanf("%d%d",&a,&b); map[a].push_back(b); map[b].push_back(a); } dfs1(1,-1,1); dfs2(1,1); build(1,n,1); while(q--) { char ch[5]; int a,b,c; scanf("%s",ch); if(ch[0]=='Q') { scanf("%d",&a); printf("%d\n",query(tid[a],1)); } else { scanf("%d%d%d",&a,&b,&c); if(ch[0]=='I') change(a,b,c); else change(a,b,-c); } } } }
时间: 2024-11-05 19:03:48