poj_3468线段树成段更新求区间和

#include<iostream>
#include<string.h>
#include<cstdio>
long long num[100010];
using namespace std;
struct st
{
    int l;
    int r;
    long long  sum;
    long long a;
} p[400100];
void build(int t,int l,int r)
{
    p[t].l=l;
    p[t].r=r;
    if(l==r-1)
    {
        p[t].sum=num[l];
        return;
    }
    int mid=(l+r)/2;
    build(t*2+1,mid,r);
    build(t*2,l,mid);
    p[t].sum=p[t*2].sum+p[t*2+1].sum;
}
void update(int t,int l,int r,int v)
{
    if(p[t].l==l&&p[t].r==r)
    {
        p[t].a += v;
		p[t].sum += v*(r -l);
		return ;
    }
    if( p[t].a )
	{
		p[2*t+1].a += p[t].a;
		p[2*t].a += p[t].a;
		p[2*t+1].sum += p[t].a*(p[2*t+1].r -p[2*t+1].l);
		p[2*t].sum += p[t].a*(p[2*t].r -p[2*t].l);
		p[t].a = 0;
	}
	int mid = (p[t].l+p[t].r)/2;
	if( l >= mid )
		update(2*t+1,l,r,v);
	else
		if( r <= mid )
			update(2*t,l,r,v);
		else
		{
			update(2*t,l,mid,v);
			update(2*t+1,mid,r,v);
		}
	p[t].sum = p[2*t].sum + p[2*t+1].sum;
}
long long query(int t,int l,int r)
{
	if(p[t].l == l && p[t].r == r )
		return p[t].sum;
	if( p[t].a )
	{
		p[2*t+1].a += p[t].a;
		p[2*t].a += p[t].a;
		p[2*t+1].sum +=p[t].a*(p[2*t+1].r - p[2*t+1].l);
		p[2*t].sum += p[t].a*(p[2*t].r - p[2*t].l);
		p[t].a = 0;
	}
	int mid =(p[t].l+p[t].r)/2;
	if( l >= mid )
		return query(2*t+1,l,r);
	else
		if( r <= mid )
			return query(2*t,l,r);
		else
			return query(2*t,l,mid) + query(2*t+1,mid,r);
}
int main()
{
    int n,q;
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        for(int i=0; i<n; i++)
            scanf("%lld",&num[i]);
        memset(p,0,sizeof(p));
        build(1,0,n+1);
        getchar();
        for(int i=0; i<q; i++)
        {
            char c;
            scanf("%c",&c);
            if(c=='Q')
            {
                int x,y;
                scanf("%d%d",&x,&y);
                getchar();
                printf("%lld\n",query(1,x-1,y));
            }
            else if(c=='C')
            {
                int x,y,z;
                scanf("%d%d%d",&x,&y,&z);
                getchar();
                update(1,x-1,y,z);
            }
        }

    }
    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值