CF444C-DZY Loves Colors【线段树,set】

1|0正题

题目链接:https://www.luogu.com.cn/problem/CF444C

---

1|1题目大意

$n$个物品第$i$个颜色为$i$，权值为$0$。要求支持$m$次操作

1. 给出$l,r,x$，对于所有区间$[l,r]$中的物品，如果颜色为$c$，那么该位置的权值加上$|c-x|$，并且颜色改为$x$
2. 询问区间权值和

---

1|2解题思路

区间染色有一种简单的做法并且可以求出每个被染色的相同颜色段。

用$set$维护每个相同的连续颜色段，那么每次修改最多会产生$3$个新的颜色端。均摊下来就是$O(n\log n)$了。

然后加一个线段树维护权值就好了，总时间复杂度也是$O(n\log n)$的

---

1|3code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#define ll long long
#define lowbit(x) (x&-x)
using namespace std;
const ll N=1e5+10;
struct node{
	ll l,r,w;
	node(ll L=0,ll rr=0,ll ww=0)
	{l=L;r=rr;w=ww;return;}
};
multiset<node> s;
ll n,m;
bool operator<(node x,node y)
{return x.r<y.r;}
struct TreeBinary{
	ll w[N<<2],lazy[N<<2];
	void Downdata(ll x,ll l,ll r){
		if(!lazy[x])return;
		ll mid=(l+r)>>1;
		w[x*2]+=lazy[x]*(mid-l+1);
		w[x*2+1]+=lazy[x]*(r-mid);
		lazy[x*2]+=lazy[x];lazy[x*2+1]+=lazy[x];
		lazy[x]=0;return;
	}
	void Change(ll l,ll r,ll val,ll L=1,ll R=n,ll x=1){
		if(L==l&&R==r){w[x]+=val*(r-l+1);lazy[x]+=val;return;}
		ll mid=(L+R)>>1;Downdata(x,L,R); 
		if(r<=mid)Change(l,r,val,L,mid,x*2);
		else if(l>mid)Change(l,r,val,mid+1,R,x*2+1);
		else Change(l,mid,val,L,mid,x*2),Change(mid+1,r,val,mid+1,R,x*2+1);
		w[x]=w[x*2]+w[x*2+1];return;
	}
	ll Ask(ll l,ll r,ll L=1,ll R=n,ll x=1){
		if(L==l&&R==r)return w[x];
		ll mid=(L+R)>>1;Downdata(x,L,R);
		if(r<=mid)return Ask(l,r,L,mid,x*2);
		if(l>mid)return Ask(l,r,mid+1,R,x*2+1);
		return Ask(l,mid,L,mid,x*2)+Ask(mid+1,r,mid+1,R,x*2+1); 
	}
}T;
signed main()
{
	multiset<node>::iterator it;
	scanf("%lld%lld",&n,&m);
	for(ll i=1;i<=n;i++)s.insert(node(i,i,i));
	while(m--){
		ll op,l,r,x;
		scanf("%lld%lld%lld",&op,&l,&r);
		if(op==1){
			scanf("%lld",&x); 
			while(1){
				it=s.lower_bound(node(l,l,l));
				node tmp=*it;
				s.erase(it);
				if(tmp.l<l&&tmp.r>r)
					T.Change(l,r,abs(x-tmp.w));
				if(tmp.l<l){
					s.insert(node(tmp.l,l-1,tmp.w));
					if(tmp.r<=r)T.Change(l,tmp.r,abs(x-tmp.w));
				}
				if(tmp.r>r){
					s.insert(node(r+1,tmp.r,tmp.w));
					if(tmp.l>=l)T.Change(tmp.l,r,abs(x-tmp.w));
					break;
				}
				if(tmp.l>=l&&tmp.r<=r)T.Change(tmp.l,tmp.r,abs(x-tmp.w));
				if(tmp.r==r)break;
			}
			s.insert(node(l,r,x));
		}
		else printf("%lld\n",T.Ask(l,r));
	}
	return 0;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/14539624.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！