更新区间求和模板

参考链接:https://blog.csdn.net/u012469987/article/details/41357377

#include<iostream>
#include<cstdio> 
using namespace std;
const int MAXN = 100000;
struct NODE{
	int left;
	int right;
	long long  sum;
	long long  lazy;
}node[MAXN*4];//四倍空间 

void pushdown(int k){
	long long c=node[k].lazy;
	if(node[k].lazy){
		node[k<<1].lazy+=c;//点np下边的两半部分的lazy需要加上点np传下来的lazy; 左半部分 
		node[k<<1|1].lazy+=c;//右半部分 
		node[k<<1].sum+=(node[k<<1].right-node[k<<1].left+1)*c;//点np下边的两半部分代表的区间总值也要加上lazy乘上区间个数 
		node[k<<1|1].sum+=(node[k<<1|1].right-node[k<<1|1].left+1)*c;//右半部分 
		node[k].lazy=0;
	}
}
//建树 
void BuildTree(int left,int right,int np){
	node[np].left=left;
	node[np].right=right;
	node[np].lazy=0;
	if(left==right){
		cin>>node[np].sum;
		return ;
	}
	int mid=(left+right)>>1;
	BuildTree(left,mid,np<<1);
	BuildTree(mid+1,right,np<<1|1);
	
	node[np].sum=node[np<<1].sum+node[np<<1|1].sum;
}
//更新 
void Update(int l,int r,int np,int c){//更新区间 l~r 内的值; l和r是不变的; 
	if(l<=node[np].left && r>=node[np].right){//如果区间l~r包含点np左区间和右区间,说明点np区间内的所有值一定要被更新; 
		node[np].sum+=(node[np].right-node[np].left+1)*c;//点np的sum增加的值为点np区间个数乘上每个点要增加的值c; 
		node[np].lazy+=c;//此时layz加上所每个点所要增加的值 
		return ;
	}
	
	pushdown(np);//如果区间l~r不包含点np左区间和右区间,那么要往下找,此时lazy的值如果不为0,则需要继续往下传; 
	
	int mid=(node[np].left+node[np].right)>>1;
	if(r<=mid){//如果需要更新的区间在np区间的左半部分,则只更新左半部分; 
		Update(l,r,np<<1,c);
	}else if(l>mid){//如果需要更新的区间在np区间的右半部分,则只更新右半部分;
		Update(l,r,np<<1|1,c);
	}else{//如果在中间,则两边都更新; 
		Update(l,r,np<<1,c);
		Update(l,r,np<<1|1,c);
	}
	
	node[np].sum=node[np<<1].sum+node[np<<1|1].sum;
}
long long ans=0;
//查询 
void findx(int l,int r,int np){
	if(l<=node[np].left && node[np].right<=r){//如果区间l~r包含点np左区间和右区间;那么直接加上np点代表的sum; 
		ans+=node[np].sum;
		return;
	}
	
	pushdown(np);//否则,继续往下查,此时,需要把上边lazy所带的值继续往下传; 
	
	int mid=(node[np].left+node[np].right)>>1;
	if(r<=mid){//此部分同上; 
		findx(l,r,np<<1);
	}else if(l>mid){
		findx(l,r,np<<1|1);
	}else{
		findx(l,r,np<<1);
		findx(l,r,np<<1|1);
	}
}
int main(){
	int n,m;
	while(cin>>n>>m){
	BuildTree(1,n,1);
	int a,b,c;
	while(m--){
		char ch[2];
		cin>>ch;
		if(ch[0]=='Q'){
			cin>>a>>b;
			ans=0;
			findx(a,b,1);
			cout<<ans<<endl;; 
		}else{
			cin>>a>>b>>c;
			Update(a,b,1,c);
		}
	}
}
	return 0;
}

 

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
 
int n, m;
struct node
{
    int l;
	int r;
    long long  sum;
	long long  mark;
}s[300000];
 
void pushup(int k)
{
    s[k].sum = s[k<<1].sum + s[(k<<1)+1].sum;
}
 
void pushdown(int k, int d)
{
    if(s[k].mark)
    {
        s[k<<1].mark += s[k].mark;
        s[(k<<1)+1].mark += s[k].mark;
        s[k<<1].sum += s[k].mark * (d - (d >> 1));
        s[(k<<1)+1].sum += s[k].mark * (d >> 1);
        s[k].mark = 0;
    }
}
 
void init(int l, int r, int k)
{
    s[k].l = l;
 	s[k].r = r;
 	s[k].mark = 0;
    if(l == r)
    {
        scanf("%I64d", &s[k].sum);
        return;
    }
    int mid = (l + r) / 2;
    init(l, mid, k << 1);
    init(mid + 1, r, (k << 1) + 1);
    s[k].sum = s[k<<1].sum + s[(k<<1)+1].sum;
}
 
void update(int l, int r, int c, int k)
{
    if(l <= s[k].l && s[k].r <= r)
    {
        s[k].mark += c;
        s[k].sum += c * (s[k].r - s[k].l + 1);
        return;
    }

    pushdown(k, s[k].r - s[k].l + 1);

    int mid = (s[k].l + s[k].r) >> 1;
    if(l <= mid) update(l, r, c, k << 1);
    if(r > mid) update(l, r, c, (k << 1) + 1);
    s[k].sum = s[k<<1].sum + s[(k<<1)+1].sum;
}

long long query(int l, int r, int k)
{
    if(l <= s[k].l && s[k].r <= r) 
		return s[k].sum;
		
    pushdown(k, s[k].r - s[k].l + 1);
 
    int mid = (s[k].l + s[k].r) / 2;
    long long res = 0;
    if(l <= mid) res += query(l, r, k << 1);
    if(r > mid) res += query(l, r, (k << 1) + 1);
    return res;
}
 
int main ()
{
    char cc;
    int a, b, c;
 
    scanf("%d%d", &n, &m);
    init(1, n, 1);
 
    for(int i = 0; i < m; i++)
    {
        scanf(" %c", &cc);
        if(cc == 'Q')
        {
            scanf("%d%d", &a, &b);
            printf("%I64d\n", query(a, b, 1));
        }
        else
        {
            scanf("%d%d%d", &a, &b, &c);
            update(a, b, c, 1);
        }
    }
 
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值