树状数组区间模板

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数数加上 xx；

2. 求出某一个数的值。

输入格式

第一行包含两个整数 NN、MM，分别表示该数列数字的个数和操作的总个数。

第二行包含 NN 个用空格分隔的整数，其中第 ii 个数字表示数列第 ii 项的初始值。

接下来 MM 行每行包含 22 或 44个整数，表示一个操作，具体如下：

操作 11： 格式：1 x y k 含义：将区间 [x,y][x,y] 内每个数加上 kk；

操作 22： 格式：2 x 含义：输出第 xx 个数的值。

输出格式

输出包含若干行整数，即为所有操作 22 的结果。

输入输出样例

输入 

5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4

输出 

6
10

 

思路

用两个数组存储，一个存储原来的数，一个存储操作的结果的和。

#include<bits/stdc++.h>
using namespace std;

int n,m;
int num[500005];
int res[500005];

int lowbit(int x){
	return x&(-x);
}

void add(int a,int b){
	while(a<=n){
		num[a]+=b;
		a+=lowbit(a);
	}
}

int query(int a){
	int ans=0;
	while(a>0){
		ans+=num[a];
		a-=lowbit(a);
	}
	return ans;
}

int main(){
	int a,b,k;
	cin>>n>>m;
	memset(num,0,sizeof(num));//存储原数以外操作的结果 
	memset(res,0,sizeof(res));//存储原来的数 
	for(int i=1;i<=n;i++){
		cin>>res[i];
	}
	while(m--){
		cin>>a; 
		if(a==1){
			cin>>a>>b>>k;
			add(a,k);
			add(b+1,-k);
		}else{
			cin>>a;
			cout << query(a)+res[a] <<endl;
		}
	}
	return 0;
}