pku3277 City Horizon

http://poj.org/problem?id=3277

线段树，离散化，成段更新

#include <stdio.h>
#include <stdlib.h>

#define lson l, m, root<<1
#define rson m+1, r, root<<1|1

const long long N = 80080;
const long long inf = (1<<31)-1;

long long max1[N<<2], color[N<<2];

long long max(long long x, long long y)
{
    return x>y? x: y;
}

void push_up(long long root)
{
    max1[root] = max(max1[root<<1], max1[root<<1|1]);
}

void push_down(long long l, long long r, long long root)
{
    color[root<<1] = max(color[root<<1], color[root]);
    color[root<<1|1] = max(color[root<<1|1], color[root]);
    color[root] = -1;
    max1[root<<1] = max(max1[root<<1], color[root<<1]);
    max1[root<<1|1] = max(max1[root<<1|1], color[root<<1|1]);
}

void build(long long l, long long r, long long root)
{
    long long m;
    color[root] = -1;
    if(l == r)
    {
        max1[root] = 0;
        return;
    }
    m = (l + r) >> 1;
    build(lson);
    build(rson);
    push_up(root);
}

void update(long long L, long long R, long long x, long long l, long long r, long long root)
{
    long long m;
    if(L <= l && r <= R)
    {
        color[root] = max(color[root], x);
        max1[root] = max(max1[root], color[root]);
        return;
    }
    m = (l + r) >> 1;
    if(color[root] != -1)
    {
        push_down(l, r, root);
    }
    if(L <= m)
    {
        update(L, R, x, lson);
    }
    if(m+1 <= R)
    {
        update(L, R, x, rson);
    }
    push_up(root);
}

long long query(long long L, long long R, long long l, long long r, long long root)
{
    long long m, result = 0;
    if(L <= l && r <= R)
    {
        return max1[root];
    }
    m = (l + r) >> 1;
    if(color[root] != -1)
    {
        push_down(l, r, root);
    }
    if(L <= m)
    {
        result = max(result, query(L, R, lson));
    }
    if(m+1 <= R)
    {
        result = max(result, query(L, R, rson));
    }
    push_up(root);
    return result;
}

struct B
{
    long long x, y;
    long long h;
}b[N>>1];

long long a[N], hash[N];

long long bs(long long l, long long r, long long x)
{
    long long m;
    while(l < r)
    {
        m = (l + r) >> 1;
        if(hash[m] == x)
        {
            return m;
        }
        if(hash[m] < x)
        {
            l = m + 1;
        }
        else
        {
            r = m;
        }
    }
}

int cmp0(const void *a, const void *b)
{
    return *(long long *)a - *(long long *)b;
}

int main()
{
    long long n, i, j, temp, result = 0;
    long long k;
    scanf("%lld", &n);
    for(i=1; i<=n; i++)
    {
        scanf("%lld%lld%lld", &b[i].x, &b[i].y, &b[i].h);
        a[(i-1)*2+1] = b[i].x;
        a[(i-1)*2+2] = b[i].y;
    }
    qsort(a+1, 2*n, sizeof(a[0]), cmp0);
    j = 1;
    hash[j] = a[1];
    for(i=2; i<=2*n; i++)
    {
        if(a[i] != a[i-1])
        {
            j = j + 1;
            hash[j] = a[i];
        }
    }
//    for(i=1; i<=j; i++)
//    {
//        printf("%5d", i);
//    }
//    printf("\n");
//    for(i=1; i<=j; i++)
//    {
//        printf("%5d", hash[i]);
//    }
//    printf("\n");
    build(1, j-1, 1);
    for(i=1; i<=n; i++)
    {
//        printf("update %lld %lld %lld\n", bs(1, j+1, b[i].x), bs(1, j+1, b[i].y)-1, b[i].h);
        update(bs(1, j+1, b[i].x), bs(1, j+1, b[i].y)-1, b[i].h, 1, j, 1);
//        for(k=1; k<=j-1; k++)
//        {
//            printf("%4d", k);
//        }
//        printf("\n");
//        for(k=1; k<=j-1; k++)
//        {
//            temp = query(k, k, 1, j, 1);
//            result += (temp * (hash[k+1]-hash[k]));
//            printf("%4d", temp);
//        }
//        printf("\n");
    }
    result = 0;
    for(i=1; i<=j-1; i++)
    {
        temp = query(i, i, 1, j, 1);
        result += (temp * (hash[i+1]-hash[i]));
//        printf("%lld %lld %lld\n", i, temp, hash[i+1]-hash[i]);
    }
    printf("%lld\n", result);
    return 0;
}