51nod 3151.小陶的疑惑 - 通过例题来初试线段树
题干
小陶最近在学数据结构,助教给他们布置了一道上机题,但是小陶不会做,你能帮帮他吗?
在一个有n个数的数列上进行m次操作,操作有两种:
1 x c: 将数列中第x个数加上c
2 x y:求出第x个数、第x+1个数、…、第y个数的和
Input
第一行两个正整数n,m 第二行n个整数,表示初始的数列 接下来m行,每行三个整数表示一个操作,格式见题面
Output
对于每个操作2输出一行,表示操作2的结果
Sample Input
输入样例1:
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
输入样例2:
2 2
0 0
1 1 1
2 1 2
输入样例3:
4 3
1 2 3 4
1 3 -1
2 3 4
2 1 2
Sample Output
输出样例1:
14
16
输出样例2:
1
输出样例3:
6
3
题解
线段树的核心在于将给定区间运用二分的方法不断分割成左右子区间,并以树型结构映射到数组存储。一般来说线段树数开四倍原数组长度。
假如一个区间在线段树数组中的下标为index,我们把下标为index * 2的空间分给该区间的左子区间,index * 2 + 1的空间分给改区间的右子区间。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll sum[800005];
ll num[200005];
int n,m;
//pushUp用来更新节点的值(通过子节点更新,比如求和,pushUp就表现为两个子节点的求和的求和)
void pushUp(int index){
sum[index] = sum[index*2] + sum[index*2+1];
}
//递归建树,l为原子数组的左界,r为原子数组的右界,index为数中节点映射到数组的下标
void build(int l,int r,int index){
if(l == r){
sum[index] = num[l];
return;
}
int m = (l + r) / 2;
//左孩子下标映射到数组中*2,右孩子映射到数组中*2+1
build(l,m,index*2);
build(m+1,r,index*2+1);
//通过pushUp来更新节点的值(由于单元素节点在递归过程中短路了
//所以pushUp调用的时候,一定是可以通过单元素节点合成新节点的
pushUp(index);
}
//更新所有有关节点的值,根据题意这里是点修改(点加)
//传入左右界,当前树数组下标,原数组中的修改点,修改值
void update(int l,int r,int currentIndex,int changeIndex,ll val){
if(l == r){
//叶子结点的情况,currentIndex指向的是树数组的下标
sum[currentIndex] += val;
return;
}
int m = (l + r) / 2;
if(changeIndex <= m){
//原数组修改点下标位于当前区间的左子区间
update(l,m,currentIndex*2,changeIndex,val);
}else{
//原数组修改点下标位于当前区间的右子区间
update(m+1,r,currentIndex*2+1,changeIndex,val);
}
//同理,叶子节点已经在递归过程中短路,调用pushUp的场合一定可以通过叶子节点合并得到更大的节点
//这里为了泛用性仍然调用了pushUp,也就是说不是单纯把遍历到的节点加上val
//而是仍然通过加了val的子节点来求和进行更新
pushUp(currentIndex);
}
//query用来查询,表现在这道题是求和,给出左右界自顶向下
//当前左界,当前右界,当前树数组下标,查询左界,查询右界
ll query(int l,int r,int index,int L,int R){
if(L <= l && r <= R){
return sum[index];
}
int m = (l + r) / 2;
ll ans = 0;
//若当前区间的左子区间和查询区间重叠
if(L <= m) ans += query(l,m,index*2,L,R);
//若当前区间的右子区间和查询区间重叠
if(R > m) ans += query(m+1,r,index*2+1,L,R);
return ans;
}
int main()
{
cin>>n>>m;
for(int i = 1 ; i <= n ; ++i) scanf("%d",num+i);
build(1,n,1);
while(m--){
int cmd;
ll l,r;
scanf("%d%lld%lld",&cmd,&l,&r);
if(cmd == 1){
//点修改的场合
update(1,n,1,l,r);
//for(int i = 1 ; i < 4 * n ; ++i) cout<<sum[i]<<' ';
}else{
//查询的场合
cout<<query(1,n,1,l,r)<<endl;
}
}
return 0;
}