线段树区间求和模板

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
struct node
{
	int l;
	int r;
	long long sum;
}tree[4000005];
long long  add[4000005];
void build(int x,int r,int rt)
{

	tree[rt].l=x;
	tree[rt].r=r;
	if(x==r)
	{
		scanf("%lld", &tree[rt].sum);

	return ;
	}
	int mid=(x+r)/2;
	build(x,mid,rt*2+1);
	build(mid+1,r,rt*2+2);
     	tree[rt].sum=tree[rt*2+1].sum +tree[rt*2+2].sum;
}
void pushdown(int rt)
{   if(add[rt])
{

	//tree[rt].sum=(tree[rt].r-tree[rt].l+1)*add[rt];
    add[rt*2+1]+=add[rt];
    add[rt*2+2]+=add[rt];
	tree[rt*2+2].sum+=(tree[rt*2+2].r-tree[rt*2+2].l+1)*add[rt];	
	tree[rt*2+1].sum+=(tree[rt*2+1].r-tree[rt*2+1].l+1)*add[rt];	
	add[rt]=0;
}
}
void update(int l,int r,int z,int rt)
{
	if(tree[rt].l>=l&&tree[rt].r<=r)
	{
		tree[rt].sum+=(tree[rt].r-tree[rt].l+1)*z;
		add[rt]+=z;
		return ;
	}
	pushdown(rt);
     int mid=(tree[rt].r+tree[rt].l)/2;
     if(l<=mid)
     {
     	update(l,r,z,rt*2+1);
     	
	 }
	 if(r>mid){
	 	update(l,r,z,rt*2+2);
	 }
	 tree[rt].sum=tree[rt*2+2].sum+tree[rt*2+1].sum;
}
long long  que(int l,int r,int rt)
{
	if(tree[rt].l==l&&tree[rt].r==r)
	{
		return tree[rt].sum;
	}
	pushdown(rt);
	long long  res=0;
	int mid=(tree[rt].l+tree[rt].r)/2;
		if(r<=mid)
		{
			res+=que(l,r,rt*2+1);
		}
		else if(l>mid)
		{
			res+=que(l,r,rt*2+2);
		}
		else
		{
			res+=que(l,mid,rt*2+1);
			res+=que(mid+1,r,rt*2+2);
		}
		return res;
}
int main()
{
	int n,q;
	while(~scanf("%d%d", &n,&q))
	{
	memset(tree,0,sizeof(tree));

	build(1,n,0);
	/*for(int i=0;i<=n;i++)
	{
		printf("%d\n", tree[i].l);
	}*/
char ch[5];
long long  x,y,z;
	while(q--)
	{
		scanf("%s", ch);
	
		if(ch[0]=='C')
	{
	
		scanf("%lld%lld%lld", &x,&y,&z);
		update(x,y,z,0); 
	}	 
	else
	{
		
		scanf("%lld%lld", &x,&y);
		long long  f=que(x,y,0);
		printf("%lld\n", f);
	}
}
}
}


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值