POJ 3468 A Simple Problem with Integers(线段树)

题意:给一串整数,每次使区段增减一个数,或者查询区段内数字和

思路:线段树模板题,用add标记。具体内容可参考代码或者刘汝佳训练指南。

#include<cstdio>
#include<cstring>
#define MAXN 100005
typedef long long LL;
LL n,Q,ql,qr,v,sumv[MAXN*3],a[MAXN],addv[MAXN*3],_sum;
char op;
void build(LL o,LL L,LL R)
{
	if(L==R) {sumv[o]=a[L];return;}
	LL M=(L+R)>>1;
	build(o*2,L,M); build(o*2+1,M+1,R);
	sumv[o]=sumv[o*2]+sumv[o*2+1];//维护根节点
}
void query(LL o,LL L,LL R,LL add)
{
	if(ql<=L && R<=qr) {_sum+=add*(R-L+1)+sumv[o];return;}
	LL M=(L+R)>>1;
	if(ql<=M)query(o*2,L,M,add+addv[o]);
	if(M<qr)query(o*2+1,M+1,R,add+addv[o]);
}
void update(LL o,LL L,LL R)
{
	if(ql<=L && R<=qr) {addv[o]+=v;sumv[o]+=v*(R-L+1);return;}
	LL M=(L+R)>>1;
	if(ql<=M) update(o*2,L,M);
	if(qr>M) update(o*2+1,M+1,R);
	sumv[o]=sumv[o*2]+sumv[o*2+1]+addv[o]*(R-L+1);
}
int main()
{
	while(~scanf("%I64d%I64d",&n,&Q))
	{
		for(LL i=1;i<=n;++i) scanf("%I64d",&a[i]);
		memset(addv,0,sizeof(addv));
		build(1,1,n);
		while(Q--)
		{
			getchar();
			op=getchar();
			if(op=='Q')
			{
				scanf("%I64d%I64d",&ql,&qr);
				_sum=0;
				query(1,1,n,0);
				printf("%I64d\n",_sum);
			}
			else
			{
				scanf("%I64d%I64d%I64d",&ql,&qr,&v);
				update(1,1,n);
			}
		}
	}
	return 0;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值