【模板】线段树洛谷 3372 线段树

最新推荐文章于 2025-04-19 22:17:07 发布

原创最新推荐文章于 2025-04-19 22:17:07 发布 · 700 阅读

0 ·

CC 4.0 BY-SA版权

c++ 同时被 2 个专栏收录

142 篇文章

订阅专栏

线段树/树状数组

24 篇文章

订阅专栏

本文介绍了一种结合线段树和Lazy标记的数据结构实现方法，用于高效处理区间加法和区间求和操作。通过具体代码示例展示了如何构建线段树、执行延迟更新以及查询区间和。

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

题目描述

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

1.将某区间每一个数加上x

2.求出某区间每一个数的和

分析

简单的线段树+lazy标记
复习一下如何写lazy。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>

using namespace std;

struct arr{
    int x,y;
    long long lazy;
    long long sum;
}f[3000000];

int n,m;

void delay(int x)//延迟标记。
{
    f[x*2].lazy+=f[x].lazy;
    f[x*2+1].lazy+=f[x].lazy;
    f[x*2].sum+=(f[x*2].y-f[x*2].x+1)*f[x].lazy;
    f[x*2+1].sum+=(f[x*2+1].y-f[x*2+1].x+1)*f[x].lazy;
    f[x].lazy=0;
}

int insert(int r,int x,int y,int add)//线段树修改操作。
{
    delay(r);
    f[r].sum+=(y-x+1)*add;
    if ((f[r].x==x)&&(f[r].y==y)) 
    {
        f[r].lazy+=add;
        return 0;
    }
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) insert(r*2,x,y,add);
        else if (x>mid) insert(r*2+1,x,y,add);
            else insert(r*2,x,mid,add),insert(r*2+1,mid+1,y,add);
}

long long find(int r,int x,int y)//线段树查找。
{
    delay(r);
    if ((f[r].x==x)&&(f[r].y==y)) return f[r].sum;
    int mid=(f[r].x+f[r].y)/2;
    if (y<=mid) return find(r*2,x,y);
        else if (x>mid) return find(r*2+1,x,y);
            else return find(r*2,x,mid)+find(r*2+1,mid+1,y);
}

int maketree(int r,int x,int y)//建树
{
    f[r].x=x; f[r].y=y; 
    f[r].lazy=0; f[r].sum=0;
    if (x==y) return 0;
    int mid=(x+y)/2;
    maketree(r*2,x,mid),maketree(r*2+1,mid+1,y);
}

int main()
{
    scanf("%d%d",&n,&m);
    maketree(1,1,n);
    for (int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        insert(1,i,i,x);
    }
    for (int i=1;i<=m;i++)
    {
        int l,x,y,w;
        scanf("%d",&l);
        if (l==1)
        {
            scanf("%d%d%d",&x,&y,&w);
            insert(1,x,y,w);
        }
        else
        {
            scanf("%d%d",&x,&y);
            printf("%lld\n",find(1,x,y));
        }
    }
    return 0;
}