我们维护区间最大值和最小值来判断区间的数是不是都相等 当更新操作更新很大的区间时 其后的更新操作(包括前一个更新区间的)复杂度就会降低 具体的证明我也不大会。。
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+100;
typedef long long ll;
struct tree{
int ma,mi,l,r;
ll val,lazy;
}T[N<<2];
void pushup(int id){
T[id].val=T[id<<1].val+T[id<<1|1].val;
T[id].ma=max(T[id<<1].ma,T[id<<1|1].ma);
T[id].mi=min(T[id<<1].mi,T[id<<1|1].mi);
}
void build(int id,int l,int r){
T[id].val=T[id].lazy=0;
T[id].l=l,T[id].r=r;
if(l==r){
T[id].mi=T[id].ma=1ll*l;
return;
}
int mid = l+r>>1;
build(id<<1,l,mid);build(id<<1|1,mid+1,r);
pushup(id);
}
void cal(int id,ll val,ll x){
T[id].lazy+=val;
T[id].ma=T[id].mi=x;
T[id].val+=1ll*(T[id].r-T[id].l+1)*val;
}
void pushdown(int id){
if(T[id].lazy){
cal(id<<1,T[id].lazy,T[id].ma);
cal(id<<1|1,T[id].lazy,T[id].ma);
T[id].lazy=0;
}
}
void update(int id,int L,int R,ll val){
if(L<=T[id].l&&R>=T[id].r){
if(T[id].ma==T[id].mi){
cal(id,abs(T[id].ma-val),val);
return;
}
}
pushdown(id);
int mid = T[id].l+T[id].r>>1;
if(L<=mid) update(id<<1,L,R,val);
if(R>mid) update(id<<1|1,L,R,val);
pushup(id);
}
ll query(int id,int L,int R){
if(L<=T[id].l&&R>=T[id].r){
return T[id].val;
}
int mid = T[id].l+T[id].r>>1;
pushdown(id);
ll ans = 0;
if(L<=mid) ans+=query(id<<1,L,R);
if(R>mid) ans+=query(id<<1|1,L,R);
return ans;
}
int main(){
int n,m,op,l,r,x;
scanf("%d%d",&n,&m);
build(1,1,n);
for(int i = 1; i <= m; i++){
scanf("%d%d%d",&op,&l,&r);
if(op==1){
scanf("%d",&x);
update(1,l,r,x);
}else{
printf("%lld\n",query(1,l,r));
}
}
return 0;
}