Codeforces Round #254 DZY Loves Colors

题意：输入n, m ; 有n给位置， 初始时第i个位置的color为i, colorfulness为0。

     有m次操作，一种是把成段的区域color更新为x， 对于更新的区域，每个位置（令第i个位置未更新前颜色为color[i]）的colorfulness增加|color[i] -x|;

     另一种操作是询问一段区间[L,R]输出该区间的colorfulness总和。

 

代码实现

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
const int N  = 111111;
int n, m, color[N<<2];
ll delta[N<<2], sum[N<<2];

void build(int l, int r, int rt)
{
    color[rt] = delta[rt] = sum[rt] = 0;
    if(l==r){
        color[rt] = l;//第i个位置颜色为i
        return ;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
}

void Up(int l, int r, int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1] + 1LL*delta[rt]*(r-l+1);
}

void Down(int rt)
{
    if(color[rt]>0)
    {
        color[rt<<1] = color[rt<<1|1] = color[rt];
        color[rt] = 0;
    }
}

void Check(int x, int l, int r,int rt)
{
    if(color[rt]>0)
    {
        delta[rt] += abs(x - color[rt]);
        sum[rt] +=  1LL *abs(x - color[rt]) * (r-l+1);
    }
    else
    {
        int m = (l+r)>>1;
        Check(x,lson); Check(x,rson);
        Up(l, r, rt);
    }
    color[rt] = x;
}

void update(int L, int R, int x, int l, int r, int rt)
{
    if(L<=l&&R>=r){
        Check(x, l, r, rt);
        return ;
    }
    Down(rt);
    int m = (l+r)>>1;
    if(L<=m) update(L, R, x, lson);
    if(R>m) update(L, R, x, rson);
    Up(l, r, rt);
}

ll query(int L, int R, int l, int r,int rt)
{
    if(L<=l&&R>=r) return sum[rt] ;
    int m = (l+r)>>1;
    ll ans = 0;
    if(R<=m)
        ans += query(L, R, lson) + 1LL * delta[rt] *(R-L+1);
    else if(L>m)
        ans += query(L, R, rson) + 1LL * delta[rt] * (R-L+1);
    else
        ans += query(L, m , lson) + query(m+1, R, rson) + 1LL * delta[rt] * (R-L+1);
    return ans;
}

int main()
{
//    freopen("in.txt", "r", stdin);
    while(scanf("%d%d", &n, &m)>0)
    {
        build(1, n, 1);
        int type, l, r, x;
        while(m--)
        {
            scanf("%d%d%d", &type, &l, &r);
            if(type == 1)
            {
                scanf("%d", &x);
                update(l, r, x, 1, n, 1);
            }
            else
                printf("%I64d\n",query(l, r, 1, n, 1));
        }
    }
    return 0;
}