F - 树状数组 2

一、题目：

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

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

2. 求出某一个数的值。

Input

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

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

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

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

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

Output

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

Sample 1

Inputcopy	Outputcopy
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

Hint

样例 1 解释：

故输出结果为 6、10。

---

数据规模与约定

对于 30%30% 的数据：�≤8N≤8，�≤10M≤10；

对于 70%70% 的数据：�≤10000N≤10000，�≤10000M≤10000；

对于 100%100% 的数据：1≤�,�≤5000001≤N,M≤500000，1≤�,�≤�1≤x,y≤n，保证任意时刻序列中任意元素的绝对值都不大于 230230。

 

二、思路：

区间修改，单点查询。

用树状数组维护差分数组，然后单点查询的时候相当于求前缀和，同时加上原数组的数。

三、代码：

#include<iostream>
using namespace std;
const int maxn = 5e5 + 5;
long long a[maxn];
long long b[maxn];
int n, m;
int lowbit(int x) {
	return x & (-x);
}
void add(int i,int x) {
	while (i <= n) {
		b[i] += x;
		i += lowbit(i);		
	}
}	
int search(int i) {
	int ans = 0;
	while (i != 0) {
		ans += b[i];
		i -= lowbit(i);
	}

	return ans;
}






int main()
{
	
	cin >> n >> m;
	int x;

	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		add(i, a[i]-a[i-1]);//差分
	}
	int t, y, k;
	while (m--) {
		cin >> t;
		if (t == 1) {
			cin >> x >> y >> k;
			add(x, k);
			add(y + 1, -k);
		}
		else {
			cin >> x;
			cout << search(x) << endl;
		
		
		}
	
	}


	return 0;
}