[Codeforces Round #254 div1] C.DZY Loves Colors 【线段树】

题目链接:CF Round #254 div1 C

 

题目分析

这道题目是要实现区间赋值的操作,同时还要根据区间中原先的值修改区间上的属性权值。

如果直接使用普通的线段树区间赋值的方法,当一个节点表示的区间完全被要求修改的区间包含时,就直接打上赋值的标记然后 return 。但是这样这个节点中每个位置原先的值不同,需要进行的属性权值修改也就不同,是不能直接实现的。如果我们在节点表示的区间被修改的区间包含时,并不直接打标记 return ,而是当节点表示的区间被修改的区间完全包含而且这个节点中的每个位置的颜色相同时,才直接打标记然后 return ,否则就继续递归下去。这样就可以直接打标记修改属性权值了。但是这样看起来是会使复杂度退化的,但是实际上通过一些我不懂的势能分析,这样修改复杂度还是 O(log n) 的。所以这样这道题就变成线段树水题了。

需要维护一下每个节点表示的区间是否颜色相同。

区间赋值的题目就可以使用这种修改方式。

 

代码

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

typedef long long LL;

inline int Abs(int x) {return x < 0 ? -x : x;}

const int MaxN = 100000 + 5;

int n, m;
int Col[MaxN * 4], Len[MaxN * 4], Pc[MaxN * 4];

LL T[MaxN * 4], D[MaxN * 4];

bool Same[MaxN * 4];

inline void Update(int x)
{
	Same[x] = Same[x << 1] && Same[x << 1 | 1] && Col[x << 1] == Col[x << 1 | 1];
	if (Same[x]) Col[x] = Col[x << 1]; 
	T[x] = T[x << 1] + T[x << 1 | 1];
	Len[x] = Len[x << 1] + Len[x << 1 | 1];
}

inline void Add(int x, LL Num)
{
	D[x] += Num;
	T[x] += (LL)Len[x] * Num;
}

inline void PushDown(int x)
{
	if (D[x] != 0)
	{
		Add(x << 1, D[x]);
		Add(x << 1 | 1, D[x]);
		D[x] = 0;
	}
	if (Pc[x] != 0)
	{
		Col[x << 1] = Pc[x << 1] = Pc[x];
		Col[x << 1 | 1] = Pc[x << 1 | 1] = Pc[x];
		Pc[x] = 0;
	}
}

void Build(int x, int s, int t)	
{	
	if (s == t)
	{
		Same[x] = true;
		Col[x] = s;
		T[x] = D[x] = Pc[x] = 0;
		Len[x] = 1;
		return;
	}
	int m = (s + t) >> 1;
	Build(x << 1, s, m);
	Build(x << 1 | 1, m + 1, t);
	Update(x); 
}

void Change(int x, int s, int t, int l, int r, int Num)
{
	if (l <= s && r >= t && Same[x])
	{
		int Temp = Abs(Col[x] - Num);
		D[x] += (LL)Temp;
		T[x] += (LL)Temp * (LL)Len[x];
		Col[x] = Pc[x] = Num;		
		return;
	}
	PushDown(x);
	int m = (s + t) >> 1;
	if (l <= m) Change(x << 1, s, m, l, r, Num);
	if (r >= m + 1) Change(x << 1 | 1, m + 1, t, l, r, Num);
	Update(x);
}

LL Query(int x, int s, int t, int l, int r)
{
	if (l <= s && r >= t) return T[x];
	PushDown(x);
	int m = (s + t) >> 1;
	LL ret = 0;
	if (l <= m) ret += Query(x << 1, s, m, l, r);
	if (r >= m + 1) ret += Query(x << 1 | 1, m + 1, t, l, r);
	return ret; 
}

int main()
{
	scanf("%d%d", &n, &m);	
	Build(1, 1, n);	
	int f, l, r, Num;
	for (int i = 1; i <= m; ++i)
	{
		scanf("%d%d%d", &f, &l, &r);
		if (f == 1) 
		{
			scanf("%d", &Num);
			Change(1, 1, n, l, r, Num);
		}
		else printf("%I64d\n", Query(1, 1, n, l, r));
	}
	return 0;
}

  

转载于:https://www.cnblogs.com/JoeFan/p/4531606.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值