洛谷P3372 线段树 1

洛谷P3372 线段树 1

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数加上 k。
2. 求出某区间每一个数的和。

输入格式

第一行包含两个整数 n, m，分别表示该数列数字的个数和操作的总个数。

第二行包含 n 个用空格分隔的整数，其中第 i 个数字表示数列第 i 项的初始值。

接下来 m 行每行包含 3 或 4 个整数，表示一个操作，具体如下：

1. 1 x y k：将区间 [x, y]内每个数加上 k。
2. 2 x y：输出区间 [x, y] 内每个数的和。

输出格式

输出包含若干行整数，即为所有操作 2 的结果。

输入输出样例

输入 #1

5 5

1 5 4 2 3

2 2 4

1 2 3 2

2 3 4

1 1 5 1

2 1 4

输出 #1

11

8

20

说明/提示

数据规模与约定

对于 30% 的数据：n≤8，m≤10。
对于 70% 的数据：n≤10^3，m≤10^4。
对于100% 的数据：1≤n,m≤10^5。

保证任意时刻数列中所有元素的绝对值之和≤10^18。

思路：线段树模板题，注意数据范围即可

代码：

#include<bits/stdc++.h>
using namespace std;

struct tt{
	int l,r;
	long long la,pre;
}t[400005];
int a[100005]={};

int len(int a){
	return t[a].r-t[a].l+1;
}
void lazy(int p){
	if(t[p].la>=1){
		t[p<<1].pre+=t[p].la*(len(p<<1));
		t[p<<1|1].pre+=t[p].la*(len(p<<1|1));
		t[p<<1].la+=t[p].la;
		t[p<<1|1].la+=t[p].la;
		t[p].la=0;
	}
}

void build(int p,int l,int r){
	t[p].l=l,t[p].r=r;
	if(l==r){
		t[p].pre=a[r];
		return ;
	}
	int mid=(l+r)>>1;
	build(p<<1,l,mid);
	build(p<<1|1,mid+1,r);
	t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}

void change(int p,int x,int y,int z){
	if(t[p].l>=x&&t[p].r<=y){
		t[p].pre+=(long long)z*(len(p));
		t[p].la+=z;
		return ;
	}
	lazy(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(x<=mid)
		change(p<<1,x,y,z);
	if(y>mid)
		change(p<<1|1,x,y,z);
	t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}

long long query(int p,int x,int y){
	if(x<=t[p].l&&y>=t[p].r){
		return t[p].pre;
	}
	lazy(p);
	int mid=(t[p].l+t[p].r)>>1;
	long long ans=0;
	if(x<=mid){
		ans+=query(p<<1,x,y);
	}
	if(y>mid){
		ans+=query(p<<1|1,x,y);
	}
	return ans;
}

int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build(1,1,n);
	while(m--){
		int x,y,k,t;
		cin>>t;
		if(t==1){
			cin>>x>>y>>k;
			change(1,x,y,k);
		}
		if(t==2){
			cin>>x>>y;
			printf("%lld\n",query(1,x,y));
		}
	}

    return 0;
}