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;}