线段树的基本操作和拓展

基本操作

一、区修区查（求和）

洛谷P3372
题目描述
如题，已知一个数列，你需要进行下面两种操作：
将某区间每一个数加上 k。
求出某区间每一个数的和。

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

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

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

1 x y k：将区间 [x, y] 内每个数加上 k。
2 x y：输出区间 [x, y] 内每个数的和。
输出格式
输出包含若干行整数，即为所有操作 2 的结果。
1 <= n, m <= 10⁵
输入输出样例
输入 #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

解析

1.线段树的基本思想：把整个区间看成一棵二叉树，把区间分成一段一段的，第一层一段，第二层两段，第三层四段。。。
2.u节点的左儿子是[u << 1]，右儿子是[u << 1 || 1]，每次更新时利用下面的节点来更新。
3.线段树需要维护的信息：
sum:考虑当前节点及子节点上的所有标记，当前区间和是多少（注意!!!不考虑所有的祖先节点）
add:lazy标记，给当前区间的所有儿子加上add

代码

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

typedef long long LL;
const int N = 100010;
int n, m;
int a[N];  //原数组
struct Node
{
   
    int l, r;  //左边界，右边界
    LL sum, add;  //区间和，lazy标记
}tr[N * 4];  //线段树需要开4倍空间

void pushup(int u)  //用当前两个子节点的信息来计算父节点的信息
{
   
    tr[u].sum = tr[u << 1].sum + tr[u << 1 | 1].sum;  //父节点的总和 = 左儿子的和+右儿子的和
}

void pushdown(int u)
{
   
    auto &root = tr[u], &left = tr[u << 1], &right = tr[u << 1 | 1];
    if (root.add)  //如果根节点上有标记
    {
   
        //left.r - left.l + 1表示区间长度
		left.add += root.add, left.sum += (LL)(left.r - left.l + 1) * root.add;
        right.add += root.add, right.sum += (LL)(right.r - right.l + 1) * root.add;
        root.add = 0;  //清空懒标记
    }
}

void build(int u, int l, int r)    //初始化
{
   
    if (l == r) tr[u] = {
   l, r, a[r], 0};
    else
    {
   
        tr[u] = {
   l, r};
        int mid = l + r >> 1;
        build(u << 1, l, mid);          //初始化左边
		build(u << 1 | 1, mid+1, r); //初始化右边
        pushup(u);
    }
}

void modify(int u, int l, int r, int d)  //修改
{
   
    if (tr[u].l >= l && tr[u].r <= r)
    {
   
        tr[u].sum += (LL)(tr[u].r - tr[u].l + 1) * d;
        tr[u].add += d;
    }
    else    //当前区间太大，一定要分裂
    {
   
        pushdown(u);  //当前区间表示的标记不能表示整个区间统一的标记，把当前区间的标记往下传
        int mid &