Austin_Chen @ 2024-12-27 19:05:23
RT,样例都没过
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=100000;
int a[N*4+10],w[N*4+10];
int lzy[N*4+10];
int n,m,x,y,k,op;
void pushup(const int u){w[u]=w[u*2]+w[u*2+1];}
void build(const int u,int L,int R)
{
if(L==R)
{
w[u]=a[L];
return;
}
int Mid=(L+R)/2;
build(u*2,L,Mid);
build(u*2+1,Mid+1,R);
pushup(u);
}
int query_o(int u,int L,int R,int p)
{
if(L==R) return w[u];
else
{
int Mid=(L+R)/2;
if(Mid>=p) query_o(u*2,L,Mid,p);
else query_o(u*2,Mid+1,R,p);
}
}
void update_o(int u,int L,int R,int p,int o)
{
if(L==R) w[u]=o;
else
{
int Mid=(L+R)/2;
if(Mid>=p) update_o(u*2,L,Mid,p,o);
else update_o(u*2+1,Mid+1,R,p,o);
pushup(u);
}
}
bool InRange(int L,int R,int l,int r){ return (l<=L)&&(R<=r);}
bool OutofRange(int L,int R,int l,int r){ return (L>r)||(R>l);}
void maketag(int u,int len,int o)
{
lzy[u]+=o;
w[u]+=len*o;
}
void pushdown(int u,int L,int R)
{
int Mid=(L+R)/2;
maketag(u*2,Mid-L+1,lzy[u]);
maketag(u*2+1,R-Mid,lzy[u]);
lzy[u]=0;
}
int query(int u,int L,int R,int l,int r)
{
if(InRange(L,R,l,r)) return w[u];
else if(!OutofRange(L,R,l,r))
{
int Mid=(L+R)/2;
return query(u*2,L,Mid,l,r)+query(u*2+1,Mid+1,R,l,r);
}
else return 0;
}
void update(int u,int L,int R,int l,int r,int o)
{
if(InRange(L,R,l,r)) maketag(u,R-L+1,o);
else if(!OutofRange(L,R,l,r))
{
int Mid=(L+R)/2;
pushdown(u,L,R);
update(u*2,L,Mid,l,r,o);
update(u*2+1,Mid+1,R,l,r,o);
pushup(u);
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
build(1,1,n);
for(int i=1;i<=m;i++)
{
cin>>op;
if(op==1)
{
cin>>x>>y>>k;
update(1,1,n,x,y,k);
}
else
{
cin>>x>>y;
cout<<query(1,1,n,x,y)<<'\n';
}
}
return 0;
}
by CSP_JAKME @ 2024-12-27 19:52:16
在query的时候也要pushdown
by Austin_Chen @ 2025-01-11 08:35:25
@CSP_JAKME谢谢,已关
by Austin_Chen @ 2025-01-11 08:38:19
@CSP_JAKME但似乎还是不对