线段树

单点修改+区间求和

考虑用lazy数组标记,当我们需要对点k所对应的区间加v时,我们不递归k的左右子树更新其内部值,而是选择lazy[k]+=v,代表这段区间已经加过v了。当我们在查询和修改中遇到一个有懒标记的点k时,由于其子树内部的值还未被更新,我们需要将懒标记的影响下传到子树中,并将懒标记清空。

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
ll n,m,a[2000010],ans[2000010],tag[2000010];
ll ls(ll x) { return x<<1; }
ll rs(ll x) { return x<<1|1; }

void lazy(ll k,ll l,ll r,ll plus)
{
	tag[k]+=plus;
	ans[k]+=plus*(r-l+1);
}

void push_up(ll k) { ans[k]=ans[ls(k)]+ans[rs(k)]; }
void push_down(ll k,ll l,ll r)
{
	ll mid=(l+r)>>1;
	lazy(ls(k),l,mid,tag[k]);
	lazy(rs(k),mid+1,r,tag[k]);
	tag[k]=0;
}

void build(ll k,ll l,ll r)
{
	tag[k]=0;
	if(l==r)
	{
		ans[k]=a[l];
		return;
	}
	ll mid=(l+r)>>1;
	build(ls(k),l,mid);
	build(rs(k),mid+1,r);
	push_up(k);
}

void up_date(ll x,ll y,ll l,ll r,ll k,ll plus)//插入 
{
	if(x<=l&&r<=y)
	{
		ans[k]+=plus*(r-l+1);
		tag[k]+=plus;
		return;
	}
	push_down(k,l,r);
	ll mid=(l+r)>>1;
	if(x<=mid) up_date(x,y,l,mid,ls(k),plus);
	if(y>mid)  up_date(x,y,mid+1,r,rs(k),plus);
	push_up(k);//回溯 
}

ll query(ll x,ll y,ll l,ll r,ll k)//查询 
{
	ll res=0;
	if(x<=l&&y>=r) return ans[k];
	ll mid=(l+r)>>1;
	push_down(k,l,r);
	if(x<=mid) res+=query(x,y,l,mid,ls(k));
	if(y>mid)  res+=query(x,y,mid+1,r,rs(k));
	return res;
}

int main()
{
	ll a1,b,c,d,e,f;
	scanf("%lld%lld",&n,&m);
	for(ll i=1;i<=n;i++) scanf("%lld",&a[i]);
	build(1,1,n);
	while(m--)
	{
		scanf("%lld",&a1);
		if(a1==1)
		{
			scanf("%lld%lld%lld",&b,&c,&d);
			up_date(b,c,1,n,1,d);
		}
		else{
			scanf("%lld%lld",&e,&f);
			printf("%lld\n",query(e,f,1,n,1));
		}
	}
	return 0;
}

区间加+区间乘+区间求和

#include<bits/stdc++.h>
using namespace std;
#define maxn 100010 
#define ll long long
#define inf 0x3f3f3f3f
struct node{
	int l,r;
	ll s[2],p[2][2];
}tr[maxn<<2];
ll a[maxn];
int n,m,p; 
void build(int x,int l,int r)
{
	tr[x].l=l;
	tr[x].r=r;
	tr[x].p[0][0]=1,tr[x].p[0][1]=0,tr[x].p[1][0]=0,tr[x].p[1][1]=1;
	if(l==r)
	{
		tr[x].s[0]=a[l];
		tr[x].s[1]=1;
		return;
	}
	int mid=l+r>>1,lc=x<<1,rc=x<<1|1;
	build(lc,l,mid);
	build(rc,mid+1,r);
	tr[x].s[0]=(tr[lc].s[0]+tr[rc].s[0])%p;
	tr[x].s[1]=r-l+1;
}
void calc1(ll a[2],ll b[2][2])
{
	ll tmp[2]={0,0};
	tmp[0]=(a[0]*b[0][0]+a[1]*b[1][0])%p;
	tmp[1]=(a[0]*b[0][1]+a[1]*b[1][1])%p;
	a[0]=tmp[0];
	a[1]=tmp[1];
}
void calc2(ll a[2][2],ll b[2][2])
{
	ll tmp[2][2];
	tmp[0][0]=(a[0][0]*b[0][0]+a[0][1]*b[1][0])%p;
	tmp[0][1]=(a[0][0]*b[0][1]+a[0][1]*b[1][1])%p;
	tmp[1][0]=(a[1][0]*b[0][0]+a[1][1]*b[1][0])%p;
	tmp[1][1]=(a[1][0]*b[0][1]+a[1][1]*b[1][1])%p;
	a[0][0]=tmp[0][0];
	a[0][1]=tmp[0][1];
	a[1][0]=tmp[1][0];
	a[1][1]=tmp[1][1];
}
void down(int x)
{
	if(tr[x].p[0][0]==1&&tr[x].p[0][1]==0&&tr[x].p[1][0]==0&&tr[x].p[1][1]==1) return;
	int mid=tr[x].l+tr[x].r>>1,lc=x<<1,rc=x<<1|1;
	calc1(tr[lc].s,tr[x].p);
	calc1(tr[rc].s,tr[x].p);
	calc2(tr[lc].p,tr[x].p);
	calc2(tr[rc].p,tr[x].p);
	tr[x].p[0][0]=1,tr[x].p[0][1]=0,tr[x].p[1][0]=0,tr[x].p[1][1]=1;
}
void add(int x,int l,int r,ll v[2][2])
{
	if(l<=tr[x].l&&tr[x].r<=r)
	{
		calc1(tr[x].s,v);
		calc2(tr[x].p,v);
		return;
	}
	down(x);
	int mid=tr[x].l+tr[x].r>>1,lc=x<<1,rc=x<<1|1;
	if(l<=mid) add(lc,l,r,v);
	if(mid<r) add(rc,l,r,v);
	tr[x].s[0]=(tr[lc].s[0]+tr[rc].s[0])%p;
}
ll query(int x,int l,int r)
{
	if(l<=tr[x].l&&tr[x].r<=r) return tr[x].s[0];
	down(x);
	int mid=tr[x].l+tr[x].r>>1,lc=x<<1,rc=x<<1|1;
	ll tmp=0;
	if(l<=mid) tmp=query(lc,l,r);
	if(mid<r) tmp+=query(rc,l,r);
	return tmp%p;
}
int main()
{
	scanf("%d%d%lld",&n,&m,&p);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int op,l,r;
		ll x;
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d%lld",&l,&r,&x);
			ll v[2][2]={x,0,0,1};
			add(1,l,r,v);
		}
		if(op==2)
		{
			scanf("%d%d%lld",&l,&r,&x);
			ll v[2][2]={1,0,x,1};
			add(1,l,r,v);
		}
		if(op==3)
		{
			scanf("%d%d",&l,&r);
			printf("%lld\n",query(1,l,r));
		}
	}
	return 0; 
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xiyuping24

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值