線段樹應用(建樹、單點更新、區間求和)

【問題描述】

給定一數列,規定有兩種操作,一是修改某個元素,二是求區間的連續和。

 

【輸入格式】

輸入數據第一行包含兩個正整數n,m(n<=100000,m<=500000)。
以下是m行,每行有三個正整數k,a,b(k=0或1、 a,b<=n)。k=0時表示將a處數字加上b,k=1時表示詢問區間[a,b]內所有數的和。

 

【輸出格式】

對於每個詢問輸出對應的答案。

【程序代碼】

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100001;
int n,m,a[MAXN],seg[MAXN*4];//線段樹數組要開大4倍

void pushup(int cur) {//依據問題,選擇若干操作
	seg[cur]=seg[cur<<1]+seg[cur<<1|1];
	//seg[cur]=min(seg[cur<<1],seg[cur<<1|1]);
}

void build(int cur,int L,int R) {//建線段樹
	if(L==R) seg[cur]=a[L];
	else {
		int mid=(L+R)/2;
		build(cur*2,L,mid);
		build(cur*2+1,mid+1,R);
		pushup(cur);
	}
}

void update(int cur,int L,int R,int x,int val) {//單點更新,將x處數字加上val
	if(L==R&&L==x) {
		seg[cur]+=val;
		return;
	}
	int mid=(L+R)/2;
	if(x<=mid)
		update(cur<<1,L,mid,x,val);
	else
		update(cur<<1|1,mid+1,R,x,val);
	pushup(cur);
}

int rangeSum(int cur,int L,int R,int x,int y) {//區間求和
	int sum=0;

	if(x>R||y<L) return 0;
	if(x<=L&&R<=y) return seg[cur];

	int mid=(L+R)/2;
	if(x<=mid) sum+=rangeSum(cur<<1,L,mid,x,y);
	if(y>mid) sum+=rangeSum(cur<<1|1,mid+1,R,x,y);

	return sum;
}

int main() {
	int a,b,c;
	cin>>n>>m;
	build(1,1,n);
	while(m--) {
		cin>>a>>b>>c;
		if(a==0) update(1,1,n,b,c);
		else cout<<rangeSum(1,1,n,b,c)<<endl;
	}
	return 0;
}

/*
in:
10 20
0 1 10
1 1 4
0 6 6
1 4 10
1 8 9
1 4 9
0 10 2
1 1 8
0 2 10
1 3 9
0 7 8
0 3 10
0 1 1
1 3 8
1 6 9
0 5 5
1 1 8
0 4 2
1 2 8
0 1 1

out:
10
6
0
6
16
6
24
14
50
41
*/



【參考文獻】

https://blog.csdn.net/guoyangfan_/article/details/81130689

https://blog.csdn.net/Floatiy/article/details/80233453

http://acm.nefu.edu.cn/JudgeOnline/problemShow.php?problem_id=1472
https://blog.csdn.net/xqx1343002589/article/details/81460713
https://blog.csdn.net/Amovement/article/details/80615499
https://blog.csdn.net/chj_zmr/article/details/9624975

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章