描述
已知一个数列,你需要进行下面两种操作:
- 将某一个数加上 x
- 求出某区间每一个数的和
输入描述
第一行包含两个正整数 n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 m 行每行包含 3 个整数,表示一个操作,具体如下:
-
1 x k
含义:将第 x 个数加上 k -
2 x y
含义:输出区间 [x,y] 内每个数的和
输出描述
输出包含若干行整数,即为所有操作 2 的结果。
样例输入 1
5 5 1 5 4 2 3 1 1 3 2 2 5 1 3 -1 1 4 2 2 1 4
样例输出 1
14 16
提示
【数据范围】
对于 30% 的数据,1≤n≤8,1≤m≤10;
对于 70% 的数据,1≤n,m≤104;
对于 100% 的数据,1≤n,m≤5×105。
数据保证对于任意时刻,a 的任意子区间(包括长度为 1 和 n 的子区间)和均在 [−231,231) 范围内。
#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[1000010];//原始数组
int c[1000010];//树状数组
int lowbit(int x){ //求二进制最低位1的权值
return x&(-x);
}
void update(int x,int k){ //单值修改,维护数组数组c
for(int i=x;i<=n;i+=lowbit(i)){
c[i]+=k;
}
}
int sum(int x){ //返回区间[1,x]的和
int ans=0;
while(x>0){
ans+=c[x];
x=x-lowbit(x);
}
return ans;
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];//相当于第i位的值从0增加了a[i]
update(i,a[i]);
}
for(int i=1;i<=m;i++){
int p,x,y;
cin>>p>>x>>y;
if(p==1){
update(x,y);
}else{ //[4,8]=[1,8]-[1,3]
cout<<sum(y)-sum(x-1)<<endl;
}
}
return 0;
}
//1 0001 1 c1=a1
//2 0010 10 c2=a1+a2
//3 0011 1 c3=a3
//4 0100 100 c4=a1+a2+a3+a4
//5 0101 1 c5=a5
//6 0110 10 c6=a5+a6
//7 0111 1 c7=a7
//8 1000 1000 c8=a1+a2+a3+a4+a5+a6+a7+a8
//
//10 1010 10 c10=a10+a9
//
//x&(-x) 可以得到对应二进制最低位1对应的权值
//
//6 00000110
//-6 11111010
// 10
//c数组中自己的父节点编号 x+lowbit(x)
//[1,4]=c4 x 0100 c4 x-=lowbit(x) 0000
//[1,6]=c6+c4 0110 c6 6-=lowbit(6) 0100 c4 0000
//[1,8]=c8 1000 c8
//[1,7]=c7+c6+c4 0111 c7 0110 c6 0100 c4 0000