[ybtoj 高效进阶 4.4] [线段树] 区间修改

[ybtoj 高效进阶 4.4] [线段树] 区间修改

题目

---

解题思路

区间修改
用懒标记传递修改的信息
如果当前被修改区间被完全包括就累加区间长度*v
并打上懒标记以便传递

区间累计
当前区间被完全包括直接返回值
否则分两段统计

---

代码

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,l,r,v,w,a[5000100],d[5000100];
long long sum[5000100];
void build(int k,int l,int r)
{
	if (l==r)
	{
	   sum[k]=a[l];
	   return;
    }
    int mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    sum[k]=sum[k*2]+sum[k*2+1];
}
void add(int k,int l,int r,int v)
{
	d[k]+=v;  //懒标记
	sum[k]+=(long long)v*(long long)(r-l+1);   //累加，修改
}
void pdown(int k,int l,int r,int mid)
{
	 if (!d[k]) return;  //懒标记没有标记过，不用传递
	 add(k*2,l,mid,d[k]);
	 add(k*2+1,mid+1,r,d[k]);  //传递，修改它的子节点
	 d[k]=0;
}
void change(int k,int l,int r,int x,int y)
{
	 if (x<=l&&r<=y)
	 {
	 	add(k,l,r,v);   //被完全包括，直接修改
	 	return;
	 } 
	 int mid=(l+r)/2;
	 pdown(k,l,r,mid);  //向下传递
	 if (x<=mid) change(k*2,l,mid,x,y);
	 if (y>mid) change(k*2+1,mid+1,r,x,y);
	 sum[k]=sum[k*2]+sum[k*2+1];
}
long long query(int k,int l,int r,int x,int y)
{
	 if (x<=l&&r<=y) return sum[k];
	 int mid=(l+r)/2;
	 long long da=0;
	 pdown(k,l,r,mid);
	 if (x<=mid) da+=query(k*2,l,mid,x,y);
	 if (y>mid) da+=query(k*2+1,mid+1,r,x,y);
	 return da;
}
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	    scanf("%d",&a[i]);
	build(1,1,n);
	for (int i=1;i<=m;i++)
	{
		scanf("%d",&w);
		if (w==1)
		{
		   scanf("%d%d%d",&l,&r,&v);
		   change(1,1,n,l,r);
		}
		else {
			scanf("%d%d",&l,&r);
			printf("%lld\n",query(1,1,n,l,r));
		}
	}
	return 0;
}