线段树模板

最新推荐文章于 2024-10-11 21:43:15 发布

Whyckck

最新推荐文章于 2024-10-11 21:43:15 发布

阅读量208

点赞数

CC 4.0 BY-SA版权

分类专栏：线段树

本文链接：https://blog.youkuaiyun.com/Whyckck/article/details/81458928

线段树专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用线段树进行区间更新与查询的方法。通过递归地构建线段树并实现区间修改和查询操作，有效地解决了区间操作问题。代码中详细展示了线段树的构建、区间更新及查询过程。

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

#include<bits/stdc++.h>
using namespace std;
#define maxn 100005
typedef long long ll;
ll n,m,arr[maxn],A[maxn<<2],add[maxn<<2];
void pushdown(ll o,ll l,ll r)  //向下传递加的数信息以及更新A数组当前节点数值
{
    if(add[o]!=0)
    {
        add[o*2] += add[o];
        add[o*2+1] += add[o];
        ll mid = l + ((r-l)>>1);
        A[o*2] += add[o] * (mid-l+1); //更新值
        A[o*2+1] += add[o] * (r-mid);
        add[o] = 0;
    }
    return ;
}
void update(ll o,ll l,ll r,ll a,ll b,ll key) //区间修改函数 a,b为要修改的区间,key为加上的值
{
    if(a<=l && b>=r)
    {
        add[o] += key;
        A[o] += key * (r-l+1);
        return ;
    }
    pushdown(o,l,r);
    ll mid = l + ((r-l)>>1);
    if(a<=mid)
        update(o*2,l,mid,a,b,key);
    if(b>=mid+1)
        update(o*2+1,mid+1,r,a,b,key);
    A[o] = A[o*2] + A[o*2+1];
    return ;
}
ll query(ll o,ll l,ll r,ll a,ll b) //查询操作
{
    if(a<=l && b>=r)
        return A[o];
    pushdown(o,l,r);
    ll mid = l + ((r-l)>>1);
    ll ans = 0;
    if(a<=mid)
        ans += query(o*2,l,mid,a,b);
    if(b>=mid+1)
        ans += query(o*2+1,mid+1,r,a,b);
    return ans;
}
void build(ll o,ll l,ll r)
{
    if(l == r)
    {
        A[o] = arr[l];
        return ;
    }
    ll mid = l + ((r-l)>>1);
    build(o*2,l,mid);
    build(o*2+1,mid+1,r);
    A[o] = A[o*2] + A[o*2+1]; //回溯维护父亲节点
    return ;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin >> n >> m;
    for(ll i = 1; i <= n; i++)
        cin >> arr[i];
    build(1,1,n);
    while(m--)
    {
        ll temp;
        cin >> temp;
        if(temp == 1)
        {
            ll a,b,k;
            cin >> a >> b >> k;
            update(1,1,n,a,b,k);
        }
        else if(temp == 2)
        {
            ll a,b;
            cin >> a >> b;
            cout << query(1,1,n,a,b) << endl;
        }
    }
    return 0;
}