蒟蒻刚学线段树,为什么RE
查看原帖
蒟蒻刚学线段树,为什么RE
308729
sheeplittlecloud楼主2021/2/2 15:50
#include<bits/stdc++.h>
#define maxn 200001
#define ll long long
using namespace std;
ll a[maxn*4+2],n,m,lasy[maxn*4+2],t[maxn*4+2];
void biuld(int x,int l,int r)
{
	if(l==r) 
	{
		t[x]=a[l];
		return ;
	}
	int mid=(l+r)/2;
	biuld(x*2,l,mid);
	biuld(x*2+1,mid+1,r);
	t[x]=t[x*2]+t[x*2+1];
}
void pushdown(int x,int l,int r,int mid)
{
	if(lasy[x])
	{
		if(mid==r)
		{
			t[x*2]+=lasy[x]*(r-l+1);
			lasy[x*2]+=lasy[x];
			cout<<lasy[x*2]<<endl;
		}
		if(l==mid+1)
		{
			t[x*2+1]+=lasy[x]*(r-l+1);
			lasy[x*2+1]+=lasy[x];
			cout<<lasy[x*2+1]<<endl;
			lasy[x]=0;
		}
	}
}
void q(int x,int l,int r,int L,int R,int k)
{
	if(L<=l&&R>=r)
	{
		t[x]+=(r-l+1)*k;
		lasy[x]+=k;
		return ;
	}
	int mid=(l+r)/2;
	pushdown(x,l,mid,mid);
	pushdown(x,mid+1,r,mid);
	if(L<=mid) q(x*2,l,mid,L,R,k);
	if(R>mid) q(x*2+1,mid+1,r,L,R,k);
	t[x]=t[x*2]+t[x*2+1];
}

ll h(int x,int l,int r,int L,int R)
{
	int mid=(l+r)/2;
	if(L<=l&&R>=r) return t[x];
	pushdown(x,l,mid,mid);
	pushdown(x,mid+1,r,mid);
	ll sum=0;
	if(L<=mid) sum+=h(x*2,l,mid,L,R);
	if(R>mid) sum+=h(x*2+1,mid+1,r,L,R);
	cout<<sum<<endl;
	return sum;
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	biuld(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int op;
		scanf("%d",op);
		if(op==1)
		{
			int x,y,k;
			cin>>x>>y>>k;
			q(1,1,n,x,y,k);
		
		}
		if(op==2)
		{
			int x,y;
			cin>>x>>y;
			int cnt=h(1,1,n,x,y);
			cout<<cnt<<endl;
		}
	}
	return 0;
}

2021/2/2 15:50
加载中...