poj 3277 City Horizon

链接：点击打开链接

求在同一水平线上的矩形面积。与HDU 1255 覆盖面积有点类似。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 40010 
int line[2*N],a[N],b[N],h[N];
struct node{
    int l;
    int r;
    int c;
    }anode[2*N*4];
void bulid(int l,int r,int n){
    anode[n].l=l;
    anode[n].r=r;
    anode[n].c=0;
    if(l+1==r)
    return;
    int mid=(l+r)>>1;
    bulid(l,mid,2*n);
    bulid(mid,r,2*n+1);
    }
void update(int id,int l,int r,int n){
    int mid;
    if(line[anode[n].l]==l&&line[anode[n].r]==r){
         if(anode[n].c<h[id])
          anode[n].c=h[id];
          return;
        }
    mid=line[(anode[n].l+anode[n].r)>>1];
    if(r<=mid)
     update(id,l,r,2*n);
     else if(l>=mid)
     update(id,l,r,2*n+1);
     else{
            update(id,l,mid,2*n);
            update(id,mid,r,2*n+1);
            }
    }
long long solve(int h,int n){
    if(anode[n].c<h)
    anode[n].c=h;
    if(anode[n].l+1==anode[n].r)
    return (long long)(line[anode[n].r]-line[anode[n].l])*anode[n].c;
    long long a=solve(anode[n].c,2*n);
    long long b=solve(anode[n].c,2*n+1);
    return a+b;
    }            
int main(){
    int i,n,j,len;
    while(~scanf("%d",&n)){
        for(i=1,j=0;i<=n;i++){
            scanf("%d %d %d",&a[i],&b[i],&h[i]);
            line[++j]=a[i];
            line[++j]=b[i];
            }
            sort(line+1,line+j+1);
            len=unique(line+1,line+j+1)-(line+1);
            bulid(1,len,1);
            for(i=1;i<=n;i++)
            update(i,a[i],b[i],1);
            long long ans=solve(0,1);
            printf("%lld\n",ans);
        }
        return 0;
    }