带懒标记线段树(区间加值,区间改值)

#include<bits/stdc++.h>
#define N 100000
using namespace std;

template<typename T>
class SegmentTree{
    public:
    T d[N],b[N],b2[N],v[N];
    void build(int s,int t,int p, const T *a);
    T getsum(int l, int r, int s, int t, int p);    	
    T getsum2 (int l, int r, int s, int t, int p);   
    void update(int l, int r, T c, int s, int t, int p);  //此更新是将[l,r]的元素加上c
    void update2(int l, int r, T c, int s, int t, int p); //此更新是将[l,r]的元素更新为c
};

template<typename T>
                                                //此处a[]无法传引用 故最好应用全局变量 以减小时空复杂度
void SegmentTree<T> :: build(int s, int t, int p,const T *a) {//如果对一整个a[]数组建立线段树:build(1,a.size(),1,a);
  b[p]=0;//对懒标记数组清空
  // 对 [s,t] 区间建立线段树,当前根的编号为 p
  if (s == t) {//单点(递归边界条件)
    d[p] = a[s];
    return;
  }
  int m = s + ((t - s) >> 1);//m即为s,t的中间值
  // 移位运算符的优先级小于加减法,所以加上括号
  // 如果写成 (s + t) >> 1 可能会超出 int 范围
  build(s, m, p * 2,a), build(m + 1, t, p * 2 + 1,a);
  // 递归对左右区间建树
  d[p] = d[p * 2] + d[(p * 2) + 1];//此处为左右区间和
}

template<typename T>
//b[i]不为0的时候,d[i]已经更新过了(但是d[i]的子节点没改变)
T SegmentTree<T> :: getsum(int l, int r, int s, int t, int p) {//在应用中查询[l,r]范围的和:getsum(l,r,1,n,1);
  // [l, r] 为查询区间, [s, t] 为当前节点包含的区间, p 为当前节点的编号
  if (l <= s && t <= r) return d[p];
  // 当前区间为询问区间的子集时直接返回当前区间的和
  int m = s + ((t - s) >> 1);
  if (b[p]) {
    // 如果当前节点的懒标记非空,则更新当前节点两个子节点的值和懒标记值
    d[p * 2] += b[p] * (m - s + 1), d[p * 2 + 1] += b[p] * (t - m);
    b[p * 2] += b[p], b[p * 2 + 1] += b[p];  // 将标记下传给子节点
    b[p] = 0;                                // 清空当前节点的标记
  }
  T sum = 0;
  if (l <= m) sum = getsum(l, r, s, m, p * 2);
  if (r > m) sum += getsum(l, r, m + 1, t, p * 2 + 1);
  return sum;
}

template<typename T>
void  SegmentTree<T> :: update(int l, int r, T c, int s, int t, int p) {//为[l,r]范围内的数增加c的值:update(l,r,c,1,n,1);
  // [l, r] 为修改区间, c 为被修改的元素的变化量, [s, t] 为当前节点包含的区间, p
  // 为当前节点的编号
  if (l <= s && t <= r) {
    d[p] += (t - s + 1) * c, b[p] += c;
    return;
  }  // 当前区间为修改区间的子集时直接修改当前节点的值,然后打标记,结束修改
  int m = s + ((t - s) >> 1);
  if (b[p] && s != t) {
    // 如果当前节点的懒标记非空,则更新当前节点两个子节点的值和懒标记值
    d[p * 2] += b[p] * (m - s + 1), d[p * 2 + 1] += b[p] * (t - m);
    b[p * 2] += b[p], b[p * 2 + 1] += b[p];  // 将标记下传给子节点
    b[p] = 0;                                // 清空当前节点的标记
  }
  if (l <= m) update(l, r, c, s, m, p * 2);
  if (r > m) update(l, r, c, m + 1, t, p * 2 + 1);
  d[p] = d[p * 2] + d[p * 2 + 1];
}

template<typename T>
void SegmentTree<T> :: update2(int l, int r, T c, int s, int t, int p) {//为[l,r]范围内的数修改为c的值:update(l,r,c,1,n,1);
  if (l <= s && t <= r) {
    d[p] = (t - s + 1) * c, b2[p] = c ,v[p] = 1;//重新计算节点值(不用算差值了)
    return;
  }
  int m = s + ((t - s) >> 1);
  // 额外数组储存是否修改值
  if (v[p]) {//只有全部修改才会打懒标记
    d[p * 2] = b2[p] * (m - s + 1), d[p * 2 + 1] = b2[p] * (t - m);
    b2[p * 2] = b2[p * 2 + 1] = b2[p];//下传懒标记
    v[p * 2] = v[p * 2 + 1] = 1;
    v[p] = 0;
  }
  if (l <= m) update2(l, r, c, s, m, p * 2);
  if (r > m) update2(l, r, c, m + 1, t, p * 2 + 1);
  d[p] = d[p * 2] + d[p * 2 + 1];
}



template<typename T>
T SegmentTree<T> :: getsum2 (int l, int r, int s, int t, int p) {
  if (l <= s && t <= r) return d[p];
  int m = s + ((t - s) >> 1);
  if (v[p]) {
    d[p * 2] = b2[p] * (m - s + 1), d[p * 2 + 1] = b2[p] * (t - m);
    b2[p * 2] = b2[p * 2 + 1] = b2[p];
    v[p * 2] = v[p * 2 + 1] = 1;
    v[p] = 0;
  }
  T sum = 0;
  if (l <= m) sum = getsum2(l, r, s, m, p * 2);
  if (r > m) sum += getsum2(l, r, m + 1, t, p * 2 + 1);
  return sum;
}

int main(){
     SegmentTree<int> tra;
     int a[10]={0,1,2,3,4,5,6,7,8,9};
    int n=9;
    tra.build(1,n,1,a);
    cout<<tra.getsum2(1,2,1,n,1)<<endl;
    cout<<tra.getsum2(1,9,1,n,1)<<endl;
    // tra.update(1,1,1,1,n,1);
    // cout<<tra.getsum(1,2,1,n,1)<<endl;
    tra.update2(1,3,1,1,n,1);
    // cout<<tra.getsum2(1,1,1,n,1)<<endl;
    tra.update2(1,3,1,1,n,1);
    cout<<tra.getsum2(1,1,1,n,1)<<endl;
    tra.update2(1,4,2,1,n,1);
    cout<<tra.getsum2(1,3,1,n,1)<<endl;
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值