线段树0分求助
查看原帖
线段树0分求助
305854
Drind楼主2021/11/2 22:53

RT,全WA了

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

struct tree
{
	int l,r,sum;
	int lazy;
} s[40000001];

int a[10000001];

void pushup(int p)
{
	s[p].sum=s[p*2].sum+s[p*2+1].sum;
}

void lazy(int p,int v)
{
	s[p].lazy+=v;
	s[p].sum+=(s[p].r-s[p].l+1)*v;
}

void pushdown(int p)
{
	lazy(p*2,s[p].lazy);
	lazy(p*2+1,s[p].lazy);
	s[p].lazy=0;
}

void update(int l,int r,int p,int v)
{
	if(l>s[p].r||r<s[p].l)
		return;
	if(l<=s[p].l&&r<=s[p].r)
	{
		s[p].sum+=(r-l+1)*v;
		s[p].lazy+=v;
		return;
	}
	pushdown(p);
	int mid=(s[p].l+s[p].r)/2;
	//if(l<=mid)
		update(l,mid,p*2,v);
	//if(r>=mid)
		update(mid+1,r,p*2+1,v); 
	pushup(p);
}

void build(int p,int l,int r)
{
	s[p].lazy=0;
	s[p].l=l;
	s[p].r=r;
	if(l==r)
	{
		s[p].sum=a[l];
		return;
	}
	int mid=(l+r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	pushup(p);
}

int query(int p,int l,int r)
{
	if(l>s[p].r||r<s[p].l)
		return 0;
	if(s[p].l>=l&&s[p].r<=r)
		return s[p].sum;
	int mid=(s[p].l+s[p].r)/2;
	pushdown(p);
	int ans=0;
	/*if(mid>r)
		ans+=query(p*2,l,r);
	else if(mid<=l)
		ans+=query(p*2+1,l,r);
	*/
	return query(p*2,l,r)+query(p*2+1,l,r);
	
}

int main() 
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int k;
		cin>>k;
		if(k==1)
		{
			int l,r,v;
			cin>>l>>r>>v;
			update(l,r,1,v);
		}
		else
		{
			int l,r;
			cin>>l>>r;
			cout<<query(1,l,r)<<endl;
		}
	}
}
2021/11/2 22:53
加载中...