目录
树状数组
应用场景:
对单个结点的修改以及对区间结果的查询
基本思想:
每个结点的深度就是其结点的二进制形式(从右到左第一个1出现的位置)
6的二进制形式中(110),从右往左数第一个1出现的位置是2(1 << 1 = 2)
7的二进制形式中(111),从右往左数第一个1出现的位置是1 (1 << 0 = 1)
8的二进制形式中(1000),从右往左数第一个1出现的位置是4(1 << 3 = 8)
为了方便 我们引入了一个 lowbit 函数,它的功能是留下第一个1其他位都变为0
lowbit(6)=2
lowbit(7)=1
因此我们规定
设节点的编号为 x ,那么这个节点的值等于 a [ x −lowbit(x)+1] 到a[x]的和
lowbit函数的实现
理论基础 :
非负数的补码和原码相同,负数的补码等于源码取反 + 1
int lowBit(int& x)
{
return x & (-x);
}
修改一个元素的值
我们需要修改所有包含e[x]这个元素的数组数组--->x+lowbit(x)
void add(int x,int v){
while (x<=len) {
e[x]+=v;
x+=lowbit(x);
}
}
查询:
也就是求 a1+a2+...+an 的值 由于e[x]=a[x-lowbit(x)]到a[x]
我们用递归来实现
int query(int x){
int sum=0;
while (x>0) {
sum+=e[x];
x-=lowbit(x);
}
return sum;
}
例题1:
题目描述
如题,已知一个数列,你需要进行下面两种操作:
1.将某一个数加上x
2.求出某区间每一个数的和
输入输出格式输入格式:
第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。
接下来M行每行包含3或4个整数,表示一个操作,具体如下:
操作1: 格式:1 x k 含义:将第x个数加上k
操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和
输出格式:输出包含若干行整数,即为所有操作2的结果。
/*
*@Author: GuoJinlong
*@Language: C++
*/
//#include <bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<string>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<list>
#include<set>
#include<iomanip>
#include<cstring>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<cassert>
#include<sstream>
#include<algorithm>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
#define ls (p<<1)
#define rs (p<<1|1)
#define mid (l+r)/2
#define over(i,s,t) for(register long long i=s;i<=t;++i)
#define lver(i,t,s) for(register long long i=t;i>=s;--i)
const int MAXN = 305;
const int INF = 0x3f3f3f3f;
const int N=5e4+7;
const int maxn=1e5+5;
const double EPS=1e-10;
const double Pi=3.1415926535897;
//inline double max(double a,double b){
// return a>b?a:b;
//}
//inline double min(double a,double b){
// return a<b?a:b;
//}
int xd[8] = {0, 1, 0, -1, 1, 1, -1, -1};
int yd[8] = {1, 0, -1, 0, -1, 1, -1, 1};
//void Fire(){
// queue<node> p;
// p.push({fx,fy,0});
// memset(fire, -1, sizeof(fire));
// fire[fx][fy]=0;
// while(!p.empty()){
// node temp=p.front();
// p.pop();
// for(int i=0;i<8;i++){
// int x=temp.x+xd[i];
// int y=temp.y+yd[i];
// if(x<0||x>=n||y<0||y>=m||fire[x][y]!=-1){
// continue;
// }
// fire[x][y]=temp.val+1;
// p.push({x,y,temp.val+1});
// }
// }
//}
//int bfs(){
// queue<node> p;
// memset(vis, 0, sizeof(vis));
// p.push({sx,sy,0});
// while (!p.empty()) {
// node temp=p.front();
// vis[temp.x][temp.y]=1;
// p.pop();
// for(int i=0;i<4;i++){
// int x=temp.x+xd[i];
// int y=temp.y+yd[i];
// if(x<0||x>=n||y<0||y>=m) continue;
// if(x==ex&&y==ey&&temp.val+1<=fire[x][y]) return temp.val+1;
// if(vis[x][y]||temp.val+1>=fire[x][y]||a[x][y]=='#') continue;
// p.push({x,y,temp.val+1});
// }
// }
// return -1;
//}
//一维哈希
//int n;
//string s;
//int bas=131;
//typedef unsigned long long ull;
//const ull mod1=100001651;
//ull a[100010];
//ull Hash(string s){
// ll ans=0;
// for(int i=0;i<s.size();i++){
// ans*=bas;
// ans+=int(s[i]);
// ans%=mod1;
// }
// return ans;
//}
//二维哈希
//using lom=unsigned long long ;
//const lom bas1=131,bas2=233;
//const int M=505;
//int n,m;
//char a[M][M];
//lom _hash[M][M];
//lom p1[M],p2[M];
//
//
//void init(){
// p1[0]=1;
// p2[0]=1;
// for(int i=1;i<=505;i++){
// p1[i]=p1[i-1]*bas1;
// p2[i]=p2[i-1]*bas2;
//
// }
//}
//void Hash(){
// _hash[0][0]=_hash[0][1]=_hash[1][0]=0;
// for(int i=1;i<=n;i++){ //前缀和
// for(int j=1;j<=m;j++){
// _hash[i][j]=_hash[i][j-1]*bas1+a[i][j]-'a';
// }
// }
// for(int i=1;i<=n;i++){ //二维前缀和
// for(int j=1;j<=m;j++){
// _hash[i][j]+=_hash[i-1][j]*bas2;
// }
// }
//
//}
//int pre[1010];
//int in[1010];
//int post[1010];
//int k;
//struct node{
// int value;
// node *l,*r;
// node (int value=0,node *l=NULL,node *r=NULL):value(value),l(l),r(r){}
//};
//void builttree(int l,int r,int &t,node * &root){
// int flag=-1;
// for(int i=l;i<=r;i++){
// if(in[i]==pre[t]){
// flag=i;
// break;
// }
// }
// if(flag==-1) return;
// root=new node(in[flag]);
// t++;
// if(flag>l) builttree(l,flag-1,t,root->l);
// if(flag<r) builttree(flag+1,r,t,root->r);
//
//}
//void preorder(node *root){
// if(root!=NULL)
// {
// post[k++]=root->value;
// preorder(root->l);
// preorder(root->r);
//
// }
//}
//void inorder(node *root){
// if(root!=NULL)
// {
// inorder(root->l);
// post[k++]=root->value;
// inorder(root->r);
//
// }
//}
//void postorder(node *root){
// if(root!=NULL)
// {
// postorder(root->l);
// postorder(root->r);
// post[k++]=root->value;
// }
//}
int n,m;
int a[10000010];
int e[10000010];
int lowbit(int x){
return x&(-x);
}
void add(int x,int k){ //修改操作
while (x<=n) {
e[x]+=k;
x+=lowbit(x);
}
}
int query(int x){ //查询操作
int sum=0;
while (x>0) {
sum+=e[x];
x-=lowbit(x);
}
return sum;
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
int x;
cin>>x;
add(i,x);
a[i]=x;
}
while (m--) {
int x;
cin>>x;
if(x==1){
int x,k;
cin>>x>>k;
add(x,k);
}
else {
int x,y;
cin>>x>>y;
cout<<query(y)-query(x-1)<<endl;
}
}
}