树状数组模板3/线段树模板2

题目描述:

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

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的结果。

样例输入:

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

样例输出:

11
8
20

提示:

建议参考题解里的模板。

一定要用long long数据类型

数据规模：

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

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

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

本题是区间修改区间询问，要求用树状数组练习一次。

时间限制: 1000ms
空间限制: 256MB

代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
struct data
{
	LL sum, lazy;
};

data tree[400005];
int a[100005];

void pushdown(int rt, int l, int r)
{
	int mid = (l + r) >> 1;
	tree[rt << 1].lazy += tree[rt].lazy;
	tree[rt << 1].sum += (mid - l + 1) * tree[rt].lazy;
	tree[rt << 1 | 1].lazy += tree[rt].lazy;
	tree[rt << 1 | 1].sum += (r - mid) * tree[rt].lazy;
	tree[rt].lazy = 0;
}

void pushup(int rt)
{
	tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;	
}

void build(int rt, int l, int r)
{
	if (l == r)
	{
		tree[rt].sum = a[l];
		return;	
	}	
	int mid = (l + r) >> 1;
	build(rt << 1, l, mid);
	build(rt << 1 | 1, mid + 1, r);
	pushup(rt);
}

void modify(int rt, int l, int r, int x, int y, int z)
{
	if (x <= l && r <= y)
	{
		tree[rt].lazy += z;
		tree[rt].sum += 1LL * z * (r - l + 1);
		return;
	}
	int mid = (l + r) >> 1;
	if (tree[rt].lazy) pushdown(rt, l, r);
	if (x <= mid) modify(rt << 1, l, mid, x, y, z);
	if (y > mid) modify(rt << 1 | 1, mid + 1, r, x, y, z);
	pushup(rt);
}

LL query(int rt, int l, int r, int x, int y)
{
	if (x <= l && r <= y) return tree[rt].sum;
	int mid = (l + r) >> 1;
	if (tree[rt].lazy) pushdown(rt, l, r);
	LL ans = 0;
	if (x <= mid) ans += query(rt << 1, l, mid, x, y);
	if (y > mid) ans += query(rt << 1 | 1, mid + 1, r, x, y);
	return ans;
}

int main(){
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	build(1, 1, n); 
	for (int i = 1; i <= m; i++)
	{
		int ope, x, y, z;
		scanf("%d", &ope);
		if (ope == 1) 
		{
			scanf("%d%d%d", &x, &y, &z);
			modify(1, 1, n, x, y, z);
		}
		else
		{
			scanf("%d%d", &x, &y);
			printf("%lld\n", query(1, 1, n, x, y));
		}
	}
}