线段树（区间操作，例题：洛谷P3372 线段树 1）

在上一节线段树（原理、构造和区间查询，例题：Balanced Lineup）中介绍了线段树的构造，下面就来说一下它的区间操作。

区间操作与Lazy-Tag有关，如果修改操作是对区间内的每个元素一一修改，就会比较繁琐低效，目前的解决办法是线段树的tree[i].data记录的是区间i的值（详细见上节），可以再定义一个tag[i],用来统一记录区间i的修改。若修改的是一个线段区间，就只进行整体上的修改，内部的每个元素先不进行修改，直到这个线段区间的一致性被破坏时，才把变化值传递给下一层的子区间，则每次区间修改的复杂度为O（）。

例题：

题目：

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

1. 将某区间每一个数加上 𝑘k。
2. 求出某区间每一个数的和。

输入格式

第一行包含两个整数 𝑛,𝑚n,m，分别表示该数列数字的个数和操作的总个数。

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

接下来 𝑚m 行每行包含 33 或 44 个整数，表示一个操作，具体如下：

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

思路：

针对这道题目，就需要运用到线段树对区间的修改和查询，就如上述一般进行区间操作。

代码：

#include<bits/stdc++.h>
using namespace std;

long long int tree[400005];
long long int tag[400005];

//求区间和 
void push_up(long long int p) {
	tree[p] = tree[p<<1] + tree[p<<1|1];
}

//构造 
void build(long long int p,long long int pl,long long int pr)
{
    if(pl == pr)
    {
        cin >> tree[p];
        return;
    }
    int mid = (pl + pr) >> 1;
    build(p<<1,pl,mid);
    build(p<<1|1,mid + 1,pr);
    tree[p] = tree[p<<1] + tree[p<<1|1];
}

//打上tag标记 
void addtag(long long int p,long long int pl,long long int pr,long long int d) {
	tag[p] += d;
	tree[p] += d * (pr - pl + 1); 
}

//将tag进行传递 
void push_down(long long int p,long long int pl,long long int pr) {
	if(tag[p]) {
		long long int mid = (pl + pr)>>1;
		addtag(p<<1,pl,mid,tag[p]);
		addtag(p<<1|1,mid + 1,pr,tag[p]);
		tag[p] = 0;
	}
}

//跟新区间，进行修改 
void update(int L,int R,int p,int pl,int pr,long long int d) {
	if(L <= pl && pr <= R) {
		addtag(p,pl,pr,d);
		return;
	}
	push_down(p,pl,pr);
	long long int mid = (pl + pr) >> 1;
	if(L <= mid) {
		update(L,R,p<<1,pl,mid,d);
	}
	if(R > mid) {
		update(L,R,p<<1|1,mid + 1,pr,d);
	}
	push_up(p);
}

//查询 
long long int query(long long int L,long long int R,long long int p,long long int pl,long long int pr) {
    if(L <= pl && R >= pr) {
    	return tree[p];
	}
	push_down(p,pl,pr);
    long long int mid = (pl + pr) >> 1;
    long long int ans=0;
    if(L <= mid) {
    	ans += query(L,R,p<<1,pl,mid);	
	}
    if(R > mid) {
    	ans += query(L,R,p<<1|1,mid + 1,pr);	
	}
    return ans;
}

int main()
{
    long long int n,m;
    cin >> n >> m;
    build(1,1,n);
    for(int i = 0; i < m; i++)
    {
        long long int j,L,R,d;
        cin >> j;
        if(j == 1) {
        	cin >> L >> R >> d;
        	update(L,R,1,1,n,d);
		}
		else{
			cin >> L >> R;
			cout << query(L,R,1,1,n) <<endl;
		}
    }
    return 0;
}