uestc数据结构训练B - 秋实大哥与花

先做的这题a不了才做的poj那题，结果是同样地原因，数组开到140000才能ac,绝对我的写法存在隐蔽的错误，这题一定要留着把正确的写法搞出来

不过这也提醒了我，比赛的时候，数组适当开大也许是明智之举

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define MAX 140100
using namespace std;

long long n,m,sum[MAX<<2],add[MAX<<2];

void pushdown(int root){
	if(root*2>=4*MAX)
			return ;
	add[root<<1]+=add[root];
	add[root<<1|1]+=add[root];
	add[root]=0;
}

void pushup(int root,int all){
	if(add[root<<1]){
		sum[root<<1]+=add[root<<1]*(all-all/2);
		pushdown(root<<1);
	}
	if(add[root<<1|1]){
		sum[root<<1|1]+=add[root<<1|1]*(all/2);
		pushdown(root<<1|1);
	}
	sum[root]=sum[root<<1]+sum[root<<1|1];
}

void build(int root,int front,int rear){
	add[root]=0;
	if(front==rear){
		scanf("%lld",&sum[root]);
		return;
	}
	int mid=(front+rear)>>1;
	build(root<<1,front,mid);
	build(root<<1|1,mid+1,rear);
	pushup(root,rear-front+1);
}

long long update(int root,int Left,int Right,int l,int r,int value){
	if(Right<l||Left>r)
			return 0;
	if(Left>=l&&Right<=r){
		add[root]+=value;
		sum[root]+=add[root]*(Right-Left+1);
		pushdown(root);
		return sum[root];
	}
	if(add[root])
		pushdown(root);
	int mid=(Right+Left)>>1;
	long long ret=update(root<<1,Left,mid,l,r,value);
	ret+=update(root<<1|1,mid+1,Right,l,r,value);
	pushup(root,Right-Left+1);
	return ret;
}

int main(){
	long long l,r,v;
	while(scanf("%lld",&n)!=EOF){
		build(1,1,n);
		scanf("%lld",&m);
		for(int i=0;i<m;i++){
			scanf("%lld %lld %lld",&l,&r,&v);
			printf("%lld\n",update(1,1,n,l,r,v));
		}
	}
	return 0;
}

过了好久才回来复习线段树，终于重写了一遍，发现问题还是在于自己逻辑思维不够严密，这次数组直接开的100010<<2，内存消耗因此小了很多，上面的版本问题在于，pushdown当root接近100000<<2的时候会re,而如果直接加上长度判断，当区间长度为1的时候返回，就没有清除root点的add值，所以正确的写法应该是下面这样的：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define MAX 100010
#define LL long long
#define lson root<<1,front,mid
#define rson root<<1|1,mid+1,rear
using namespace std;

struct Node{
	LL value,add;
}tree[MAX<<2];
int n,m;

void pushdown(int root,int l){//l可能等于1 
	if(l<=1){
		tree[root].add=0;
		return;
	}
	tree[root<<1].add+=tree[root].add;
	tree[root<<1|1].add+=tree[root].add;
	tree[root].add=0;
}

void pushup(int root,int l){
	tree[root<<1].value+=tree[root<<1].add*(l-l/2);
	pushdown(root<<1,l-l/2);
	tree[root<<1|1].value+=tree[root<<1|1].add*(l/2);
	pushdown(root<<1|1,l/2);
	tree[root].value=tree[root<<1].value+tree[root<<1|1].value;
}

void build(int root,int front,int rear){
	if(front==rear){
		scanf("%lld",&tree[root].value);
		tree[root].add=0LL;
		return;
	}
	int mid=(front+rear)>>1;
	build(lson);
	build(rson);
	pushup(root,rear-front+1);
}

LL update(int root,int front,int rear,int l,int r,int v){
	if(front>=l&&rear<=r){
		tree[root].add+=v;
		tree[root].value+=(rear-front+1)*tree[root].add;
		pushdown(root,rear-front+1);//pushdown 的作用就是把root所带的add下方到子节点，因此无论root的长度是多少，该功能一定要成功 。 
		return tree[root].value;
	}
	pushdown(root,rear-front+1);
	int mid=(front+rear)>>1;
	LL ret=0LL;
	if(mid>=l){
		ret+=update(lson,l,r,v);
	}
	if(mid+1<=r){
		ret+=update(rson,l,r,v);
	}
	pushup(root,rear-front+1);//pushup 的作用就是由两个子节点的值求root节点的值，故要考虑子节点的add有没有被消化这两种情况  调用pushup时，根节点的长度一定大于1 
	return ret;
}

int main(){
	while(~scanf("%d",&n)){
		build(1,1,n);
		scanf("%d",&m);
		int a,b;
		LL c;
		for(int i=0;i<m;i++){
			scanf("%d %d %lld",&a,&b,&c);
			printf("%lld\n",update(1,1,n,a,b,c));
		}
	}
}