单点修改+区间求和
考虑用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;
}