1. 程式人生 > >數列操作 — 線段樹入門

數列操作 — 線段樹入門

# 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;
}