洛谷P3372 【模板】线段树 1（树状数组解法）

题目描述

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

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

2.求出某区间每一个数的和

输入输出格式

输入格式：

 

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

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

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

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

操作2： 格式：2 x y 含义：输出区间[x,y]内每个数的和

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

输出样例#1： 

11
8
20

说明

时空限制：1000ms,128M

数据规模：

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

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

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

（数据已经过加强^_^，保证在int64/long long数据范围内）

解题思路

原数组a[i]等于差分数组b[]的前i项之和（差分相关见上一篇博客）

所以a[1] = b[1]， a[2] = b[1] + b[2]， a[r] = b[1] + b[2] + ...+ b[r]。

所以sum = a[1] + a[2] +...+ a[r]  = (b[1]) + (b[1] + b[2]) +... + (b[1] + b[2] + ... + b[r])

               = r*b[1] + (r-1)*b[2] + ... + (r-(r-1))b[r]  

               = r*(b[1] + b[2] + ... + b[r]) - (0*b[1] + 1*b[2] + ... + (r-1)*b[r])

所以，维护两个树状数组，一个维护b[i]（tree1[]），一个维护b[i]*(i - 1)（tree2[]）。

当给区间[x,y]的b[i]增加k时，就相当于b[x]+=k，b[y+1]-=k，b[x]*(x - 1)增加了k*(x-1)，b[y+1]*y增加了-k*y

区间[x,y]的和为sum_y(前y个数的和) - sum_x_1（前x-1个和）

代码如下

#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
ll tree1[100005];
ll tree2[100005];
int n, m;
int lowbit(int x)
{
	return x & (-x);
}
void update(int x, int k, ll tree[])
{
	for(int i = x; i <= n; i += lowbit(i))
		tree[i] += k;
}
ll query(int x, ll tree[]) 
{
	ll sum = 0;
	for(int i = x; i > 0; i -= lowbit(i))
		sum += tree[i];
	return sum;
}
int main()
{	
	while(cin >> n >> m){
		memset(tree1, 0, sizeof(tree1));
		memset(tree2, 0, sizeof(tree2));
		ll x, y;
		x = 0;
		for(int i = 1; i <= n; i ++){
			scanf("%d", &y);
			ll w = y - x;
			update(i, w, tree1);
			update(i, w * (i - 1), tree2);
			x = y;
		}
		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, k, tree1);
				update(x, k * (x - 1), tree2);
				update(y + 1, -k, tree1);
				update(y + 1, -k * y, tree2);
			}
			else {
				int x, y;
				scanf("%d%d", &x, &y);
				ll q1 = y * query(y, tree1) - query(y, tree2);
				ll q2 = (x - 1) * query(x - 1, tree1) - query(x - 1, tree2);
				printf("%lld\n", q1 - q2);
			}
		}
	}
	return 0;
}