POJ 3277 City Horizon 扫描线+离散化+线段树

题意的大体意思就是求N个矩形的面积并。可以算做是基础的模板题。

第一次用线段树写扫描线，感觉还不是很难。对于扫描线，选择扫描的方式很重要，第一次写的是从左往右扫，不知道是不是因为某处想到不周到还是因为其他的原因，总之一直在TLE，后来换成从上往下扫，跑了250MS，而且思路也清晰了许多。

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

#define LL long long
#define Max(a,b) (a) > (b) ? (a) : (b)
#define Min(a,b) (a) < (b) ? (a) : (b)
#define EPS (1e-8)
#define Left 1
#define Right -1;

using namespace std;

struct L
{
    LL l,r,h;
}line[41000];

LL X[82000],area;

bool cmp(LL a,LL b)
{
    return a < b;
}

bool cmph(L l1,L l2)
{
    return l1.h > l2.h;
}

LL delsame(LL *p,LL n)
{
    if(n == 0)
        return 0;
    LL i,top;
    for(i = 1,top = 1;i < n; ++i)
    {
        if(p[i] != p[i-1])
            p[top++] = p[i];
    }
    return top;
}

LL bs(LL *p,LL n,LL m)
{
    LL s = 0,e = n-1,mid;
    while(s < e)
    {
        mid = (s+e)>>1;
        if(p[mid] == m)
            return mid;
        if(m < p[mid])
            e = mid-1;
        else
            s = mid+1;
    }
    mid = (s+e)>>1;
    if(p[mid] == m)
        return mid;
    return -1;
}

struct ST
{
    LL l,r;
    bool mark,ls,rs;
}st[324000];

void init_st(LL site,LL l,LL r)
{
    if(l >= r)
        return;
    if(l == r-1)
    {
        st[site].l = l;
        st[site].r = r;
        st[site].mark = false;
        st[site].ls = false;
        st[site].rs = false;
        return ;
    }
    st[site].l = l;
    st[site].r = r;
    st[site].mark = false;
    st[site].ls = false;
    st[site].rs = false;

    LL mid = (l+r)>>1;
    init_st(site<<1,l,mid);
    init_st(site<<1|1,mid,r);
}

void Cal_Area(LL site,LL l,LL r,LL h)
{
    if(l >= r || st[site].mark)
        return ;
    if(st[site].l == l && st[site].r == r)
    {
        if(st[site].ls == false && st[site].rs == false)
        {
            st[site].mark = true;
            area += h*(X[r]-X[l]);
            return ;
        }
        else
        {
            LL mid = (st[site].l+st[site].r)>>1;
            st[site].ls = true;
            st[site].rs = true;
            Cal_Area(site<<1,l,mid,h);
            Cal_Area(site<<1|1,mid,r,h);
            if(st[site].ls && st[site].rs && st[site<<1].mark && st[site<<1|1].mark)
            {
                st[site].mark = true;
            }
        }
    }

    LL mid = (st[site].l+st[site].r)>>1;

    if(r <= mid)
    {
        st[site].ls = true;
        Cal_Area(site<<1,l,r,h);
        if(st[site].ls && st[site].rs && st[site<<1].mark && st[site<<1|1].mark)
        {
            st[site].mark = true;
        }
        return ;
    }
    if(mid <= l)
    {
        st[site].rs = true;
        Cal_Area(site<<1|1,l,r,h);
        if(st[site].ls && st[site].rs && st[site<<1].mark && st[site<<1|1].mark)
        {
            st[site].mark = true;
        }
        return ;
    }
    st[site].ls = true;
    st[site].rs = true;
    Cal_Area(site<<1,l,mid,h);
    Cal_Area(site<<1|1,mid,r,h);
    if(st[site].ls && st[site].rs && st[site<<1].mark && st[site<<1|1].mark)
    {
        st[site].mark = true;
    }
}

int main()
{
    LL i,n,top,s1,s2;
    while(scanf("%lld",&n) != EOF)
    {
        for(top = 0,i = 0;i < n; ++i)
        {
            scanf("%lld %lld %lld",&line[i].l,&line[i].r,&line[i].h);
            X[top++] = line[i].l;
            X[top++] = line[i].r;
        }

        sort(X,X+top,cmp);
        sort(line,line+n,cmph);

        top = delsame(X,top);


        init_st(1,0,top-1);

        for(area = 0,i = 0;i < n; ++i)
        {
            s1 = bs(X,top,line[i].l);
            s2 = bs(X,top,line[i].r);

            Cal_Area(1,s1,s2,line[i].h);
        }
        printf("%lld\n",area);
    }
    return 0;
}