UESTC 1057 秋实大哥与花 线段树模板 区间更新+查询

秋实大哥与花

Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

Submit  Status

秋实大哥是一个儒雅之人，昼听笙歌夜醉眠，若非月下即花前。

所以秋实大哥精心照料了很多花朵。现在所有的花朵排成了一行，每朵花有一个愉悦值。

秋实大哥每天要对着某一段连续的花朵歌唱，然后这些花朵的愉悦值都会增加一个相同的值vv(vv可能为负)。

同时他想知道每次他唱完歌后这一段连续的花朵的愉悦值总和是多少。

Input

第一行有一个整数nn，表示花朵的总数目。

第二行包含nn个整数aiai，表示第ii朵花初始的愉悦值。

第三行包含一个整数mm，表示秋实大哥唱了mm天的歌。

接下来mm行，每行包含三个整数ll rr vv，表示秋实大哥对着[l,r][l,r]这个区间内的花朵歌唱，每朵花的愉悦值增加了vv。

1≤n，m，ai，|v|≤1000001≤n，m，ai，|v|≤100000，1≤l≤r≤n。1≤l≤r≤n。

Output

输出共mm行，第ii行表示秋实大哥完成第ii天的歌唱后，那一段花朵的愉悦值总和。

Sample input and output

Sample Input	Sample Output
3 0 0 0 3 1 2 1 1 2 -1 1 3 1	2 0 3

#include <bits/stdc++.h>

using namespace std;

const int inf=0x3f3f3f3f;
const int maxn=100005;
typedef long long ll;

int n,a[maxn],q;

struct node{
	int l,r;
	ll sum,lazy;
	void update(ll x){
		//整个区间上的数都加x 
		sum+=1LL*(r-l+1)*x;//1LL用来强制转换成long long
		lazy+=x; 
	}
}tree[maxn*4];

void push_up(int x){//向上对父亲 
	tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum;
}

void push_down(int x){//向下对儿子的操作 
	int lazyval=tree[x].lazy;
	if(lazyval){
		tree[x<<1].update(lazyval);
		tree[x<<1|1].update(lazyval);
		tree[x].lazy=0;
	}
} 

void build(int x,int l,int r){
	tree[x].l=l;
	tree[x].r=r;
	tree[x].sum=tree[x].lazy=0;
	if(l==r)
		tree[x].sum=a[l];
	else{
		int mid=(l+r)/2;
		build(x<<1,l,mid);
		build(x<<1|1,mid+1,r);
		push_up(x);
	}
}

void update(int x,int l,int r,ll val){
	int L=tree[x].l,R=tree[x].r;
	if(l<=L&&R<=r)//包含tree[x]
		tree[x].update(val);
	else{
		push_down(x); 
		int mid=(L+R)/2;
		if(mid>=l) update(x<<1,l,r,val);
		if(r>mid)	update(x<<1|1,l,r,val);
		push_up(x);
	}
}

ll query(int x,int l,int r){
	int L=tree[x].l,R=tree[x].r;
	if(l<=L&&R<=r) 
		return tree[x].sum;
	else{
		push_down(x);
		ll ans=0;
		int mid=(L+R)/2;
		if(mid>=l)	ans+=query(x<<1,l,r);
		if(r>mid)	ans+=query(x<<1|1,l,r);
		push_up(x);
		return ans;
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	build(1,1,n);//建树 
	scanf("%d",&q);
	for(int i=1;i<=q;i++){
		int l,r,val;
		scanf("%d%d%d",&l,&r,&val);
		update(1,l,r,val);
		printf("%lld\n",query(1,l,r));
	}
	return 0;
}