线段树基础模板 + 解释

线段树解决的三个问题

（1）更新点，查询区间。

（2）更新区间，查询点。

（3）更新区间，查询区间。

处理一段连续区间的插入,查找,统计,查询等操作。
复杂度：当有n个元素时，对区间操作的复杂度为O(logn)级别。

线段树是一种完美二叉树，每个结点记录一段区间信息。

对于每个非叶节点[a,b)，令mid=(a+b)/2,则其左右儿子结点代表的区间为[a,mid),[mid,b);根节点下标为1，非根节点下标为p，其左右子节点的下标分别为p*2和p*2+1.
注：由于线段树的性质可知，线段树需要的空间是所需要处理的最长线段的两倍多，所以需要申请3-4倍大小的空间，建议直接申请四倍。

基本操作代码：

一.更新点

1.建树

//建树
void build(int p, int l, int r) // p为根结点，l左端点，r右端点
{
    if(l == r)
    {
        tree[p] = a[l];
        return;
    }

    int mid = (l + r) >> 1;
    build(p << 1, l, mid);
    build(p << 1 | 1, mid + 1, r);

    tree[p] = tree[p << 1] + tree[p << 1 | 1]; // 父节点的值等于左右儿子结点值的和 
}

// 更新值
void change(int p, int l, int r, int x, int num)
{
    if(l == r) // 如果是叶子结点，则对p进行操作
    {
        tree[p] += num; 
        return;
    }
    int mid = (l + r) >> 1;
    if(x <= mid) change(p << 1, l, mid, x, num);
    else
        change(p << 1 | 1, mid + 1, r, x, num);

    tree[p] = tree[p << 1] + tree[p << 1 | 1];
}

//对区间进行操作
int find(int p, int l, int r, int x, int y)
{
    if(x <= l && r <= y)
        return tree[p];
    int mid = (l + r) >> 1;

    if(y <= mid)
        return find(p << 1, l, mid, x, y);
    if(x > mid)
        return find(p << 1 | 1, mid + 1, r, x, y);

    return (find(p << 1, l, mid, x, mid) + find(p << 1 | 1, mid + 1, r, mid + 1, y));

}

二.更新区间

1.lazy标记下放

void Add(int p, int l, int r, int num)
{
    add[p] += num;
    tree[p] += (long long)(r - l + 1) * num;
}

void pushdown(int p, int l, int r, int mid)
{
    if(add[p] == 0)
        return;
    Add(p << 1, l, mid, add[p]);
    Add(p << 1 | 1, mid + 1, r, add[p]);
    add[p] = 0;
}

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

    tree[p] = tree[p << 1] + tree[p << 1 | 1];
}

void change(int p, int l, int r, int x, int y, int num)
{
    if(l >= x && r <= y)
    {
        return Add(p, l, r,num);
    }
    int mid = (l + r) >> 1;
    pushdown(p, l, r, mid);
    if(x <= mid)
        change(p << 1, l, mid, x, y, num);
    if(y > mid)
        change(p << 1 | 1, mid + 1, r, x, y, num);
    tree[p] = tree[p << 1] + tree[p << 1 | 1];
}

long long query(int p, int l, int r, int x, int y)
{
    if(l >= x && r <= y)
        return tree[p];
    int mid = (l + r) >> 1;
    long long res = 0;
    pushdown(p, l, r, mid);

    if(x <= mid)
        res += query(p << 1, l, mid, x, y);
    if(y > mid)
        res += query(p << 1 | 1, mid + 1, r, x, y);
    return res;
}

2.标记永久性

void build(int p, int l, int r)
{
    if(l == r)
    {
        tree[p] = a[l];
        return;
    }
    int mid = (l + r) >> 1;

    build(p << 1, l, mid);
    build(p << 1 | 1, mid + 1, r);
    tree[p] = tree[p << 1] + tree[p << 1 | 1];
}

void change(int p, int l, int r, int x, int y, int num)
{

    if(l >= x && r <= y)
    {
        add[p] += num;
        return;
    }

    tree[p] += (long long)(min(r, y) - max(l, x) + 1) * num;
    int mid = (l + r) >> 1;

    if(x <= mid)
        change(p << 1, l, mid, x, y, num);
    if (y > mid)
        change(p << 1 | 1, mid + 1, r, x, y, num);

}

long long query(int p, int l, int r, int x, int y)
{
    if(l >= x && r <= y)
        return tree[p] + (long long)(r - l + 1) * add[p];
    int mid = (l + r) >> 1;
    long long res = (long long)(min(r, y) - max(l, x) + 1) * add[p];
    if(x <= mid)
    {
        res += query(p << 1, l, mid, x, y);
    }
    if(y > mid)
        res += query(p << 1 | 1, mid + 1, r, x, y);
    return res;
}