Codeforces 444C DZY Loves Colors(线段树)



Codeforces 444C DZY Loves Colors

题意：

给出n个节点，每个节点有个颜色值和权值，初始时，1~n个节点的颜色值为别为1~n，权值为0。

有两种操作

1是修改区间(l,r)内所有节点的颜色值为x，并且节点的权值增加的值为节点原先的颜色值减去x后的绝对值。

2是输出区间(l,r)内所有节点的权值的和。

分析：

线段树区间操作。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
using namespace std;
const int maxn=100005;
struct nd{
	long long cnt,sum,col;
}node[4*maxn];
int n,m,x,a,b,v;
void init(int root,int l,int r){
	node[root].sum=0;
	node[root].cnt=0;
	if(l==r){
		node[root].col=l;
		return;
	}
	node[root].col=0;
	int mid=(l+r)/2;
	init(root*2+1,l,mid);
	init(root*2+2,mid+1,r);
	return;
}
void update(int root,int l,int r){
	if(b<l||a>r)return;
	if(a<=l&&b>=r&&node[root].col){
		node[root].cnt+=abs(node[root].col-v);
		node[root].sum+=abs(node[root].col-v)*(r-l+1);
		node[root].col=v;
		return;
	}
	int mid=(l+r)/2;
	if(node[root].cnt){
		node[root*2+1].col=node[root].col;
		node[root*2+2].col=node[root].col;
		node[root*2+1].cnt+=node[root].cnt;
		node[root*2+2].cnt+=node[root].cnt;
		node[root*2+1].sum+=node[root].cnt*(mid-l+1);
		node[root*2+2].sum+=node[root].cnt*(r-mid);
		node[root].cnt=0;
	}
	update(root*2+1,l,mid);
	update(root*2+2,mid+1,r);
	node[root].sum=node[root*2+1].sum+node[root*2+2].sum;
	if(node[root*2+1].col==node[root*2+2].col)node[root].col=node[root*2+1].col;
	else node[root].col=0;
	return;
}
long long count(int root,int l,int r){
	if(b<l||a>r)return 0;
	if(a<=l&&b>=r)return node[root].sum;
	int mid=(l+r)/2;
	if(node[root].cnt){
		node[root*2+1].col=node[root].col;
		node[root*2+2].col=node[root].col;
		node[root*2+1].cnt+=node[root].cnt;
		node[root*2+2].cnt+=node[root].cnt;
		node[root*2+1].sum+=node[root].cnt*(mid-l+1);
		node[root*2+2].sum+=node[root].cnt*(r-mid);
		node[root].cnt=0;
	}
	long long ans=count(root*2+1,l,mid)+count(root*2+2,mid+1,r);
	return ans;
}
int main(){
	scanf("%d%d",&n,&m);
	init(0,1,n);
	for(int i=0;i<m;i++){
		scanf("%d",&x);
		if(x==1){
			scanf("%d%d%d",&a,&b,&v);
			update(0,1,n);
		}
		else{
			scanf("%d%d",&a,&b);
			printf("%I64d\n",count(0,1,n));
		}
	}
	return 0;
}