求助,线段树炸了,只有30分
查看原帖
求助,线段树炸了,只有30分
61737
z1212楼主2021/4/11 17:11
#include<bits/stdc++.h>
#define ll long long
#define int long long
using namespace std;
const int m1=4e5+5;
long long a[m1],ans[m1],ad[m1],n,m,q,b,c,d;
inline void add(ll k,ll l,ll r,ll v) {
	ad[k]+=v;
	ans[k]+=(r-l+1)*v;
}
void build(ll k,ll l,ll r) {
	if(l==r) {
		ans[k]=a[l];
		return;
	}
	ll mid=(l+r)>>1;
	build(k*2,l,mid);
	build(k*2+1,mid+1,r);
	ans[k]=ans[k*2]+ans[k*2+1];
}
void pushdown(ll k,ll l,ll r,ll mid) {
	if(ad[k]==0)return;
	add(k*2,l,mid,ad[k]);
	add(k*2+1,l,mid,ad[k]);
	ad[k]=0;
}
void modify(ll k,ll l,ll r,ll x,ll y,ll v) {
	if(l>=x&&r<=y)return add(k,l,r,v);
	ll mid=(l+r)>>1;
	pushdown(k,l,r,mid);
	if(x<=mid)modify(k*2,l,mid,x,y,v);
	if(mid<y)modify(k*2+1,mid+1,r,x,y,v);
	ans[k]=ans[k*2+1]+ans[k*2];
}
long long query(ll k,ll l,ll r,ll x,ll y) {
	if(l>=x&&r<=y)return ans[k];
	ll mid=(l+r)>>1;
	long long res=0;
	pushdown(k,l,r,mid);
	if(x<=mid)res+=query(k*2,l,mid,x,y);
	if(mid<y)res+=query(k*2+1,mid+1,r,x,y);
	return res;
}
signed main() {
	freopen("P3372_4","r",stdin);
	freopen("1.out","w",stdout);
	cin>>n>>m;
	for(int i=1; i<=n; i++)
		scanf("%lld",&a[i]);
	build(1,1,n);
	while(m--) {
		scanf("%lld",&q);
		switch(q) {
			case 1: {
				scanf("%lld%lld%lld",&b,&c,&d);
				modify(1,1,n,b,c,d);
				break;
			}
			case 2: {
				scanf("%lld%lld",&b,&c);
				cout<<query(1,1,n,b,c)<<endl;
				break;
			}
		}
	}

	return 0;
}
2021/4/11 17:11
加载中...