數列操作 — 線段樹入門
阿新 • • 發佈:2019-02-15
# include <cstdio> # define N 500000 // 利用二叉樹的父子節點關係來儲存,因此必須保證陣列開的足夠大 struct node{ int l,r,v; // left、right、value ;左右孩子由二叉樹性質計算。 }; node st[N]; int a[N]; void build(int v,int l,int r){ st[v].l = l; st[v].r = r; if (l == r) { // 是葉子,直接賦值,跳出 st[v].v=a[l]; return ; } int mid = (l+r)/2; // 不是葉子,往下擴充套件 build(2*v,l,mid); build(2*v+1,mid+1,r); st[v].v=st[v*2].v+st[v*2+1].v; // 擴展出孩子後才能計算value域 } void insert(int v,int w,int p){ // 在以v為根的樹中,尋找節點w,並把節點w的值加上p 。 格外注意 某些程式語言中insert不能作為識別符號 if (w>=st[v].l && w<=st[v].r){ st[v].v += p; // 父子之間為統治關係,凡是經過的節點都要更新value } if (st[v].l==st[v].r) return; // 找到終點,跳出 int mid = (st[v].l+st[v].r)/2; if (w<=mid) { // 判斷左右孩子 insert(v*2,w,p); }else { insert(v*2+1,w,p); } } int getsum(int v,int l,int r){ // 返回以v為根節點的樹中,區間 [l,r] 的value域 if (st[v].l==l && st[v].r==r){ return st[v].v; // 找到,返回 } int mid = (st[v].l+st[v].r)/2; if (r<=mid) { return getsum(v*2,l,r); // 區間 [l,r] 在v的左枝上 }else if (l>mid) { return getsum(v*2+1,l,r); // 右枝 }else { // 兵分兩路 return getsum(v*2,l,mid) + getsum(v*2+1,mid+1,r); } } int main(void){ freopen("1080.in","r",stdin); int n; scanf("%d",&n); for (int i(1);i<=n;i++) { scanf("%d",&a[i]); } build(1,1,n); int m,t,a,b; for (scanf("%d",&m);m;m--){ scanf("%d%d%d",&t,&a,&b); if (t & 1) { insert(1,a,b); }else { printf("%d\n",getsum(1,a,b)); } } return 0; }