线段树 Raw code

最新推荐文章于 2024-06-25 19:01:03 发布

转载最新推荐文章于 2024-06-25 19:01:03 发布 · 76 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/js2xxx/p/7663665.html

本文详细介绍了线段树懒惰传播的实现原理及应用，通过具体的代码示例展示了如何构建线段树、更新节点值以及查询区间和。适用于需要高效处理区间更新和查询问题的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


#include <iostream>
#include <cstdio>

using namespace std;

struct seg_tree
{
    int l, r;
    long long v, lazy;
} tree[400001];

void build(int k, int l, int r)
{
    tree[k].l = l, tree[k].r = r;
    if (l == r)
    {
        scanf("%lld", &tree[k].v);
    }
    else
    {
        int m = (l + r) >> 1;
        build(k << 1, l, m);
        build(k << 1 | 1, m + 1, r);
        tree[k].v = tree[k << 1].v + tree[k << 1 | 1].v;
    }
}

void busify(int k)
{
    int l = tree[k << 1].l, r = tree[k << 1].r;
    tree[k << 1].lazy += tree[k].lazy;
    tree[k << 1].v += (r - l + 1) * tree[k].lazy;

    l = tree[k << 1 | 1].l, r = tree[k << 1 | 1].r;
    tree[k << 1 | 1].lazy += tree[k].lazy;
    tree[k << 1 | 1].v += (r - l + 1) * tree[k].lazy;
    tree[k].lazy = 0;
}

void update(int k, int l, int r, long long v)
{
    if (l > tree[k].r || r < tree[k].l) return;
    if (tree[k].l >= l && tree[k].r <= r)
    {
        tree[k].lazy += v;
        tree[k].v += (tree[k].r - tree[k].l + 1) * v;
    }
    else
    {
        if(tree[k].lazy) busify(k);
        int m = (tree[k].l + tree[k].r) >> 1;
        if (l <= m) update(k << 1, l, r, v);
        if (r > m) update(k << 1 | 1, l, r, v);
        tree[k].v = tree[k << 1].v + tree[k << 1 | 1].v;
    }
}

long long sum(int k, int l, int r)
{
    if (l > tree[k].r || r < tree[k].l) return 0;
    if (tree[k].l >= l && tree[k].r <= r)
    {
        return tree[k].v;
    }
    else
    {
        if(tree[k].lazy) busify(k);
        int m = (tree[k].l + tree[k].r) >> 1; long long s = 0;
        if (l <= m) s += sum(k << 1, l, r);
        if (r > m) s += sum(k << 1 | 1, l, r);
        return s;
    }
}

int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    build(1, 1, n);
    while (m--)
    {
        int op, a, b; long long c;
        scanf("%d", &op);
        if (op == 1)
        {
            scanf("%d%d%lld", &a, &b, &c);
            update(1, a, b, c);
        }
        else if (op == 2)
        {
            scanf("%d%d", &a, &b);
            printf("%lld\n", sum(1, a, b));
        }
    }
}

转载于:https://www.cnblogs.com/js2xxx/p/7663665.html