模板求调
查看原帖
模板求调
525152
huwanpeng楼主2023/9/22 21:16
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define ll long long
using namespace std;
int n,m;
ll a[100010],add[400010],sum[400010];//懒惰标记add 
void push_up(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	push_up(rt);
}
void pushdown(int rt,int ln,int rn)//ln:左子树元素个数  rn:右子树元素个数 
{
	if(add[rt])
	{
		add[rt<<1]+=add[rt];
		add[rt<<1|1]+=add[rt];
		sum[rt<<1]+=add[rt]*ln;
		sum[rt<<1|1]+=add[rt]*rn;
		add[rt]=0;
	}
}
void update(int L,int R,ll k,int l,int r,int rt)
{
	if(L<=l&&r<=R)
	{
		sum[rt]+=k*(r-l+1);
		add[rt]+=k;
		return;
	}
	if(l==r) return;
	int mid=(l+r)>>1;
	pushdown(rt,mid-l+1,r-mid);
	if(L<=mid)	update(L,R,k,l,mid,rt<<1);
	if(R>mid) update(L,R,k,mid+1,R,rt<<1|1);
	push_up(rt);
}
ll query(int L,int R,int l,int r,int rt)
{
	ll ans=0;
//	printf("%d %d\n",l,r);
	if(L<=l&&r<=R)
	{
		return sum[rt];
	}
	if(l==r) return 0;
	int mid=(l+r)>>1;
	pushdown(rt,mid-l+1,r-mid);
	if(L<=mid) ans+=query(L,R,l,mid,rt<<1);
	if(R>mid) ans+=query(L,R,mid+1,r,rt<<1|1);
	return ans;
}
int main()
{
//	freopen("P3372.in","r",stdin);
//	freopen("P3372.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	build(1,n,1);
	while(m--)
	{
		int op,x,y,k;
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d%lld",&x,&y,&k);
			update(x,y,k,1,n,1);
		}
		else if(op==2)
		{
			scanf("%d%d",&x,&y);
			printf("%lld\n",query(x,y,1,n,1));
		}
	}
	return 0;
}

线段树模板,学校老师教的什么鬼,答案查询都没讲

2023/9/22 21:16
加载中...