CSP-S 例题12-1 线段树的操作

描述

输入 n, m表初始有 n 个数, 接下来 m 行输入, Q x y 表示询问区间 [x, y]的和；C x y z 表示区间 [x, y] 内所有数加上 z

输入描述

共有m+2行，第一行输入两个整数n,m。
第二行输入n个数，接下来的m行输入具体的操作。

输出描述

[x,y]的和

样例输入 1 

10 5 1 2 3 4 5 6 7 8 9 10 Q 4 4 Q 1 10 Q 2 4 C 3 6 3 Q 2 4

样例输出 1 

4 55 9 15

#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
struct node{
	int l,r; //区间的左右边界 
	long long s; //区间的和 
	long long add;  //延时标记，用在区间修改 
}tree[maxn*4];
int a[maxn]; //原数组
int n,m;
void build(int p,int l,int r){ //节点编号  区间左边界  区间右边界 
	tree[p].l=l;
	tree[p].r=r;
	tree[p].add=0;
	if(l==r){
		tree[p].s=a[l];
		return;
	} 
	int mid=(l+r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	tree[p].s=tree[p*2].s+tree[p*2+1].s;
	return; 
} 
void push(int i){ //处理第i号节点的延时标记 
 	if(tree[i].add!=0){
 		tree[i*2].s += (tree[i*2].r-tree[i*2].l+1)*tree[i].add; //更改左节点的区间和 
 		tree[i*2+1].s += (tree[i*2+1].r-tree[i*2+1].l+1)*tree[i].add; //更改右节点的区间和 
 		tree[i*2].add+=tree[i].add;
		tree[i*2+1].add+=tree[i].add;
		tree[i].add=0; 
	} 
}
void change(int i,int l,int r,int d){
	if(l<=tree[i].l && r>=tree[i].r){ //判断是否完全包括第i个节点 
		tree[i].s += (tree[i].r-tree[i].l+1) * d;
		tree[i].add += d; //记录延时标记
		return; 
	}
	push(i);
	int mid=(tree[i].l+tree[i].r)/2;
	if(l<=mid){
		change(i*2,l,r,d);
	} 
	if(r>mid){
		change(i*2+1,l,r,d);
	} 
	tree[i].s=tree[i*2].s+tree[i*2+1].s;
	return;
}      
long long ask(int i,int l,int r){
	if(l<=tree[i].l && r>=tree[i].r){ //判断是否完全包括第i个节点 
		return tree[i].s;
	}
	push(i);
	int mid=(tree[i].l+tree[i].r)/2;
	long long value=0;
	if(l<=mid){
		value += ask(i*2,l,r);
	}
	if(r>mid){
		value += ask(i*2+1,l,r);
	}
	return value;
} 
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,n);
	while(m--){
		char ch;
		int x,y,d;
		cin>>ch;
		if(ch=='Q'){
			cin>>x>>y;
			cout<<ask(1,x,y)<<endl;
		}else{
			cin>>x>>y>>d;
			change(1,x,y,d);
		}
	}
	return 0;
}