牛客小白月赛20-E(线段树)

E-区区区间

#include<stdio.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=2e5+5;
ll a[N];
struct Tree {
	int l,r;
	ll dat,lazy;
} t[N*4];
void build(int p,int l,int r) {
	t[p].l=l,t[p].r=r;
	if(l==r) {
		t[p].dat=a[l];
		return;
	}
	int mid=(l+r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	t[p].dat=t[p*2].dat+t[p*2+1].dat;
}
void spread(int p) {
	if(t[p].lazy) {
		int mid=(t[p].l+t[p].r)/2;
		t[p*2].lazy=t[p].lazy;
		t[p*2+1].lazy=t[p].lazy+mid-t[p].l+1;
		t[p*2].dat=(t[p*2].lazy*2+(mid-t[p].l))*(mid-t[p].l+1)/2;
		t[p*2+1].dat=((t[p*2+1].lazy)*2+(t[p].r-mid-1))*(t[p].r-mid)/2;
		t[p].lazy=0;
	}
}
void change(int p,int l,int r,int k) {
	if(l<=t[p].l&&t[p].r<=r) {
		t[p].lazy=k+t[p].l-l;
		t[p].dat=(t[p].lazy*2+t[p].r-t[p].l)*(t[p].r-t[p].l+1)/2;
		return;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)/2;
	if(l<=mid) {
		change(p*2,l,r,k);
	}
	if(r>mid) {
		change(p*2+1,l,r,k);
	}
	t[p].dat=t[p*2].dat+t[p*2+1].dat;
}
ll ask(int p,int l,int r) {
	if(l<=t[p].l&&t[p].r<=r) {
		return t[p].dat;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)/2;
	ll ans=0;
	if(l<=mid) {
		ans+=ask(p*2,l,r);
	}
	if(r>mid) {
		ans+=ask(p*2+1,l,r);
	}
	return ans;
}
int main() {
	int n,m;
	scanf("%d %d",&n,&m);
	for(int i=1; i<=n; ++i)scanf("%lld",&a[i]);
	build(1,1,n);
	while(m--) {
		int cut,x,y,z;
		scanf("%d %d %d",&cut,&x,&y);
		if(cut==1) {
			scanf("%d",&z);
			change(1,x,y,z);
		} else {
			printf("%lld\n",ask(1,x,y));
		}
	}
	return 0;
}

区间更新，等差数列求和公式，不难，但是写的有点麻烦，又是调了很长时间的bug，线段树写的不熟练，多练吧。