CF Round #254 (Div. 1) C. DZY Loves Colors

我们维护区间最大值和最小值来判断区间的数是不是都相等  当更新操作更新很大的区间时 其后的更新操作(包括前一个更新区间的)复杂度就会降低 具体的证明我也不大会。。

#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;
}
原创文章 85 获赞 103 访问量 2471

猜你喜欢

转载自blog.csdn.net/weixin_43824564/article/details/106150544
今日推荐