线段树模板/区间查改+标记延迟

仅为模板,仅供参考。
下面是模板代码:

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N = 100005;
int mx = 0;
ll su[N];
struct Node
{
    int l, r;
    ll sum, add;
}t[N*4];
void build(int p, int l, int r)//建树
{
    mx = max(mx, p);
    t[p].l = l; t[p].r = r;
    if (l == r) {t[p].sum = su[l]; return;}
    int mid = (l +r) >> 1;
    build((p << 1), l, mid);
    build((p << 1) | 1, mid + 1, r);
    t[p].sum = t[p<<1].sum + t[(p << 1)|1].sum;
    return;
}
void spread(int p)//标记延迟
{
    int l = p * 2;
    int r = p * 2 + 1;
    if (t[p].add)
    {
        t[l].sum += t[p].add * (t[l].r - t[l].l + 1);
        t[r].sum += t[p].add * (t[r].r - t[r].l + 1);
        t[l].add += t[p].add;
        t[r].add += t[p].add;
        t[p].add = 0;
    }
}
void change(int p, int l, int r, ll x)
{
    if (l <= t[p].l && r >= t[p].r)
    {
        t[p].sum += x * (t[p].r - t[p].l + 1);
        t[p].add += x;
        return;
    }
    spread(p);
    int mid = (t[p].r + t[p].l) >> 1;
    if (l <= mid) change(p << 1, l, r, x);
    if (r > mid) change((p << 1) | 1, l, r, x);
    t[p].sum = t[p << 1].sum + t[(p << 1) | 1].sum;
}
ll ask(int p, int l, int r)
{
   // cout << l << " " <<r << " " << p << endl;
    if (l <= t[p].l && r >= t[p].r) return t[p].sum;
    spread(p);
    int mid = (t[p].l + t[p].r) >> 1;
    ll val = 0;
    if (l <= mid) val += ask(p << 1, l, r);
    if (r > mid) val += ask((p << 1) | 1, l, r);
    return val;
}
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
        scanf("%lld", &su[i]);
    build(1, 1, n);
    //for (int i = 1; i <= mx; i++)
      //  cout << t[i].l << " " << t[i].r <<" " <<t[i].sum << endl;

    while(m--)
    {
        int op;
        scanf("%d", &op);
        if(op == 1)
        {
            int l, r;
            ll k;
            scanf("%d%d%lld", &l, &r, &k);
            change(1, l, r, k);//区间改(l-r区间都+k)
          //  for (int i = 1; i <= mx; i++)
               // cout << t[i].l << " " << t[i].r << " " << t[i].sum << " " << t[i].add <<endl;
        }
        else
        {
            int l, r;
            scanf("%d%d", &l, &r);//区间查(l-r的和)
           // for (int i = 1; i <= mx; i++)
            //    cout << t[i].l << " " << t[i].r << " " << t[i].sum << " " << t[i].add <<endl;
            printf("%lld\n", ask(1, l, r));
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值