【樹鏈剖分】樹上的詢問
阿新 • • 發佈:2018-12-11
【題目描述】
給你一棵具有N個點(編號為1到N)M條邊的樹,並給定各個點權的值,然後有3種操作: I C1 C2 K:把C1與C2的路徑上的所有點權值加上K D C1 C2 K:把C1與C2的路徑上的所有點權值減去K Q C:查詢節點編號為C的權值
【輸入格式】
輸入包含多組測試資料。 對於每組測試資料的第一行包含三個整數N,M,P,分別表示N個結點(1<=N<=50000),M條邊(M=N-1)和P(1≤P≤100000)個操作。 接下來一行為N個整數A1,A2,...AN(0≤Ai≤1000),表示一開始每個結點的值。 接下來的M行,每行兩個整數u和v,表示u到v有一條邊。 接下來P行,表示相應的操作,其中0<=k<=1000。
【輸出格式】
對於每個詢問輸出一行,為詢問結點的值。
樹鏈剖分的裸題,不過對區間的操作比較簡單,我們可以直接用樹狀陣列的區間修改與單點查詢,沒必要寫線段樹,程式碼上會簡短不少
#include<iostream> #include<cstdio> #include<queue> #include<iomanip> #include<cstring> #include<algorithm> #include<bitset> using namespace std; int n,m,q,a[50005],h[50005],cnt,size[50005],prt[50005],son[50005],deep[50005],seg[50005],top[50005],rev[50005],tree[50005]; struct edge { int to,next; }w[100005]; void add(int x,int y) { cnt++; w[cnt].to=y; w[cnt].next=h[x]; h[x]=cnt; } void dfs(int x,int fa) { prt[x]=fa; size[x]=1; deep[x]=deep[fa]+1; for(int i=h[x];i;i=w[i].next) { int to=w[i].to; if(to==fa) continue; dfs(to,x); size[x]+=size[to]; if(size[to]>size[son[x]]) son[x]=to; } } void dfs2(int x) { if(son[x]) { top[son[x]]=top[x]; seg[son[x]]=++seg[0]; rev[seg[0]]=son[x]; dfs2(son[x]); } for(int i=h[x];i;i=w[i].next) { int to=w[i].to; if(top[to]) continue; top[to]=to; seg[to]=++seg[0]; rev[seg[0]]=to; dfs2(to); } } void add2(int x,int k) { // cout<<x<<endl; while(x<=n) { tree[x]+=k; x+=(x&-x); } } int sum(int x) { int ans=0; while(x) { ans+=tree[x]; x-=(x&-x); } return ans; } void uprange(int x,int y,int z) { // cout<<x<<" "<<y<<" "<<z<<endl; while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); add2(seg[top[x]],z); add2(seg[x]+1,-z); x=prt[top[x]]; } if(deep[x]>deep[y]) swap(x,y); add2(seg[x],z); add2(seg[y]+1,-z); } int main() { while(~scanf("%d%d%d",&n,&m,&q)) { memset(tree,0,sizeof(tree)); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=m;i++) { int x,y; scanf("%d%d",&x,&y); add(x,y); add(y,x); } dfs(1,0); seg[0]=seg[1]=rev[1]=top[1]=1; dfs2(1); for(int i=1;i<=n;i++) add2(seg[i],a[i]),add2(seg[i]+1,-a[i]); for(int i=1;i<=q;i++) { char s[10]; scanf("%s",s); if(s[0]=='Q') { int x; scanf("%d",&x); printf("%d\n",sum(seg[x])); } else if(s[0]=='D') { int l,r,z; scanf("%d%d%d",&l,&r,&z); uprange(l,r,-z); } else { int l,r,z; scanf("%d%d%d",&l,&r,&z); uprange(l,r,z); } } } }