P3368 模板【树状数组】2

题目描述

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

将某区间每一个数数加上 xxx；

求出某一个数的值。

输入格式

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

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

接下来 MMM 行每行包含 222 或 444个整数，表示一个操作，具体如下：

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

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

输出包含若干行整数，即为所有操作 222 的结果。
输入输出样例
输入 #1

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

输出 #1

6
10

说明/提示
样例 1 解释：

故输出结果为 6、10。
数据规模与约定

对于 30%30%30% 的数据：N≤8,M≤10N\le8,M\le10N≤8,M≤10；

对于 70%70%70% 的数据：N≤10000,M≤10000N\le 10000,M\le10000N≤10000,M≤10000；

对于 100%100%100% 的数据：1≤N,M≤5000001 \leq N, M\le 5000001≤N,M≤500000，1≤x,y≤n1 \leq x, y \leq n1≤x,y≤n，保证任意时刻序列中任意元素的绝对值都不大于 2302^{30}230。

涉及到区间修改，可以用差分+树状数组。

#include <iostream>
#include <algorithm>
using namespace std;

typedef long long ll;
const int maxn = 5e5 + 10;
ll t[maxn];
int a[maxn];
int n, m;

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

void update(int i, int val) {
	for(; i <= n; i += lowbit(i))
		t[i] += val;//, cout << i << " " << t[i] << " " << a[i] << endl; 
}

void update_len(int l, int r, int val) {
	for(int i = l; i <= r; i += lowbit(i)) {
		cout << i << " " << t[i] << " ";
		t[i] += val * min(lowbit(i), i - l + 1);
		cout << t[i] << endl; 
	}
}

ll query(int i) {
	ll ans = 0;
	for(; i >= 1; i -= lowbit(i)) {
		ans += t[i];
	}
	return ans;
}

int main() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++)	cin >> a[i], update(i, a[i] - a[i - 1]);
	int id, x, y, z;
//	for(int i = 1; i <= n; i++) {
//		cout << t[i] << " ";
//	}
//	cout << endl;
	for(int i = 1; i <= m; i++) {
		cin >> id;
		if(id == 1) {
			cin >> x >> y >> z;
			update(x, z);
			update(y + 1, -z);
//			for(int j= 1; j <= n; j++)
//				cout << t[j] << " ";
//			cout << endl;
		}
		else {
			cin >> x;
			cout << query(x) << endl;
		}
	}
	return 0;
}