POJ3468 线段树求和(线段树模板2)

原题链接:POJ3468

解析:本题也是线段树的一种用法,算得上裸题,我用这题来加深对线段树的理解,以及熟悉线段树的模板。郭炜老师的代码是演示指针写法,但我更喜欢用数组来表示,所以改成了我喜欢的模板。

PS:刚刚刚刚刚才知道,这种方法是延迟标记,也就是说当对一端区间的叶子结点进行相同操作的时候,不要急着全部完成,不然递归等操作需要很多,我们可以记录对这个区间将要做什么操作,等查询的时候将所有对此区间的操作一次做完,这样就省的一次一次遍历了。举个例子,就比如对a进行+1操作,进行一千亿亿次,结果和+一千亿亿是一样的,然而效率却天差地别。

失误点:Query函数里的第一行,tree[root].l == s && tree[root].r == e,第一次将tree[root].r 写成tree[root].l,导致答案错误。

代码示例:

#include<iostream>
#include<cstdio>
using namespace std;
long long ans;
struct Node{
	int l,r;
	long long Inc;
	long long sum;
	int Mid(){
		return (l+r)/2;
	}
}tree[400050];
void BuildTree(int root,int l,int r){
	tree[root].l = l;
	tree[root].r = r;
	tree[root].sum = 0;
	tree[root].Inc = 0;
	if(l == r)	return;
	int mid = (l+r)/2;
	BuildTree(2*root+1,l,mid);
	BuildTree(2*root+2,mid+1,r);
}
void Insert(int root,int i,int v){
	if(tree[root].l == i && tree[root].r == i){
		tree[root].sum = v;
		return;
	}
	tree[root].sum += v;
	if(i <= tree[root].Mid())	Insert(2*root+1,i,v);
	else Insert(2*root+2,i,v);
}
void Add(int root,int s,int e,long long c){
	if(tree[root].l == s && tree[root].r == e){
		tree[root].Inc += c;
		return;
	}
	tree[root].sum += c*(e-s+1);
	if(e <= tree[root].Mid())	Add(2*root+1,s,e,c);
	else if(s > tree[root].Mid())	Add(2*root+2,s,e,c);
	else{
		Add(2*root+1,s,tree[root].Mid(),c);
		Add(2*root+2,tree[root].Mid()+1,e,c);
	}
}
long long Query(int root,int s,int e){
	if(tree[root].l == s && tree[root].r == e){
		return tree[root].sum + tree[root].Inc*(e-s+1);
	}
	tree[root].sum += tree[root].Inc*(tree[root].r - tree[root].l+1);
	Add(2*root+1,tree[root].l,tree[root].Mid(),tree[root].Inc);
	Add(2*root+2,tree[root].Mid()+1,tree[root].r,tree[root].Inc);
	tree[root].Inc = 0;
	if(e <= tree[root].Mid())	return Query(2*root+1,s,e);
	else if(s > tree[root].Mid())	return Query(2*root+2,s,e);
	else{
		return Query(2*root+1,s,tree[root].Mid()) +
				Query(2*root+2,tree[root].Mid()+1,e); 
	}
}
int main()
{
	int n,q,h;
	scanf("%d%d",&n,&q);
	BuildTree(0,1,n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&h);
		Insert(0,i,h);
	}
	char c[2];
	for(int i = 0;i < q;i++){
		scanf("%s",c);
		if(c[0] == 'Q'){
			int s,e;
			scanf("%d%d",&s,&e);
			printf("%lld\n",Query(0,s,e));
		}else{
			int s,e,c;
			scanf("%d%d%d",&s,&e,&c);
			Add(0,s,e,c);	
		}
	}
	return 0;
}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

迷亭1213

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值