洛谷P3368 【模板】树状数组 2（差分）

 

题目描述

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

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

2.求出某一个数的值

输入输出格式

输入格式：

 

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

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

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

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

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

 

输出格式：

 

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

 

输入输出样例

输入样例#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

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

解题思路

先是差分的概念。比如说数组a[]是 1 3 2 4 1（从下标1开始），那么a[]的差分数组b[] (b[i] = a[i] - a[i-1]，a[0]设为0) 是1，2，-1，2，-3。

所以a[i] = b[i] + a[i-1] = b[i] + b[i-1] + a[i - 2]。

因为a[1] = b[1],所以a[i] = b[i] + b[i-1] + b[i-2] + ....+b[1]。

建立维护b[]的树状数组，当区间a[]的区间[x,y]加上k时，就相当于b[x]+=k，b[y+1]-=k，修改这两个点就可以了，a[i]的值就是前i个b[]的和。

代码如下

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int a[500005];
int b[500005];
int c[500005];
int lowbit(int x)
{
	return x&(-x);
}
void update(int x, int n, int k)
{
	b[x] += k;
	for(int i = x; i <= n; i += lowbit(i))
		c[i] += k;
}
int query(int x)
{
	int sum = 0;
	for(int i = x; i > 0; i -= lowbit(i))
		sum += c[i];
	return sum;
}
int main()
{
	int n, m;
	while(cin >> n >> m){
		memset(a, 0, sizeof(a));
		memset(c, 0, sizeof(c));
		for(int i = 1; i <= n; i ++){
			scanf("%d", &a[i]);
			int x = a[i] - a[i - 1];
			update(i, n, x);
		}
		for(int i = 0; i < m; i ++){
			int ins;
			scanf("%d", &ins);
			if(ins == 1){
				int x, y, k;
				scanf("%d%d%d", &x, &y, &k);
				update(x, n, k);
				update(y + 1, n, -k);
			}
			else {
				int x;
				cin >> x;
				printf("%d\n", query(x));
			}
		}
	}
	return 0;
}