线段树模板

最新推荐文章于 2024-07-13 23:05:26 发布

袁子力

最新推荐文章于 2024-07-13 23:05:26 发布

阅读量101

点赞数 2

文章标签：算法

版权属于CSDN-Yuanzili2

本文链接：https://blog.csdn.net/Yuanzili2/article/details/132190908

版权

#include <iostream>
#include <cstring>

using namespace std;

const int MAX_N = 1e5 + 5;

int n;
int a[MAX_N], tree[MAX_N << 2], lazy[MAX_N << 2]; // 线段树和懒标记

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

// 更新p结点的区间和
void push_up(int p)
{
    tree[p] = tree[p << 1] + tree[p << 1 | 1];
}

// 下传lazy标记
void push_down(int p, int len)
{
    if (lazy[p])
    {
        lazy[p << 1] += lazy[p];
        lazy[p << 1 | 1] += lazy[p];
        tree[p << 1] += (len - (len >> 1)) * lazy[p];
        tree[p << 1 | 1] += (len >> 1) * lazy[p];
        lazy[p] = 0;
    }
}

// 线段树的区间修改操作
void update(int ql, int qr, int add, int l, int r, int p)
{
    if (ql <= l && qr >= r) // 完全包含，直接修改
    {
        tree[p] += add * (r - l + 1);
        lazy[p] += add;
        return;
    }
    push_down(p, r - l + 1);
    int mid = (l + r) >> 1;
    if (ql <= mid)
        update(ql, qr, add, l, mid, p << 1);
    if (qr > mid)
        update(ql, qr, add, mid + 1, r, p << 1 | 1);
    push_up(p);
}

// 线段树的区间查询操作
int query(int ql, int qr, int l, int r, int p)
{
    if (ql <= l && qr >= r)
        return tree[p];
    push_down(p, r - l + 1);
    int res = 0, mid = (l + r) >> 1;
    if (ql <= mid)
        res += query(ql, qr, l, mid, p << 1);
    if (qr > mid)
        res += query(ql, qr, mid + 1, r, p << 1 | 1);
    return res;
}

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    build(1, n, 1);
    update(1, 3, 2, 1, n, 1);  // 区间[1, 3]加2
    update(2, 4, 3, 1, n, 1);  // 区间[2, 4]加3
    cout << query(1, 3, 1, n, 1) << endl;  // 查询区间[1, 3]的和
    cout << query(2, 5, 1, n, 1) << endl;  // 查询区间[2, 5]的和
    return 0;
}