线段树爆0,求救
  • 板块P2357 守墓人
  • 楼主yinbe
  • 当前回复2
  • 已保存回复3
  • 发布时间2023/12/2 13:37
  • 上次更新2023/12/2 15:33:29
查看原帖
线段树爆0,求救
759152
yinbe楼主2023/12/2 13:37
#include<iostream>
using namespace std;
struct tree
{
	int l,r;
	long long sum,add;
	#define l(x) t[x].l
	#define r(x) t[x].r
	#define sum(x) t[x].sum
	#define add(x) t[x].add
}t[800005];
int a[200005],n,m;
void build(int p,int l,int r)
{
	l(p)=l,r(p)=r;
	if(l==r)
	{
		sum(p)=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	sum(p)=sum(p*2)+sum(p*2+1);
	return;
}
void spread(int p)
{
	if(add(p))
	{
		sum(p*2)+=add(p)*(r(p*2)-l(p*2)+1);
		sum(p*2+1)+=add(p)*(r(p*2+1)-l(p*2+1)+1);
		add(p*2)+=add(p);
		add(p*2+1)+=add(p);
		add(p)=0;
	}
	return;
}
void change(int p,int l,int r,int d)
{
	if(l<=l(p)&&r>=r(p))
	{
		sum(p)+=(long long)(d)*(r(p)-l(p)+1);
		add(p)+=d;
		return;
	}
	spread(p);
	int mid=(l(p)+r(p))>>1;
	if(l<=mid)change(p*2,l,r,d);
	if(r>mid)change(p*2+1,l,r,d);
}
long long ask(int p,int l,int r)
{
	if(l<=l(p)&&r>=r(p))
	{
		return sum(p);
	}
	spread(p);
	int mid=(l(p)+r(p))>>1;
	long long 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()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	build(1,1,n);
	while(m--)
	{
		int flag;
		scanf("%d",&flag);
		if(flag==1)
		{
			int l,r,d;
			scanf("%d%d%d",&l,&r,&d);
			change(1,l,r,d);
		}
		else if(flag==2)
		{
			int d;
			scanf("%d",&d);
			change(1,1,1,d);
		}
		else if(flag==3)
		{
			int d;
			scanf("%d",&d);
			change(1,1,1,-d);
		}
		else if(flag==4)
		{
			int l,r;
			scanf("%d%d",&l,&r);
			printf("%lld\n",ask(1,l,r));
		}
		else if(flag==5)
		{
			printf("%lld\n",ask(1,1,1));
		}
	}
	return 0;
}
2023/12/2 13:37
加载中...