【HDU 1542】Atlantis（线段树+离散化，矩形面积并)

#include<algorithm>
#include<cstdio>
#include<cstring>
#define dd double
#define ll long long
#define N 201
using namespace std;
struct P{dd s,e,h;int f;}p[N];
struct Tree{dd sum;int c;}t[N<<5];
dd a,b,c,d,x[N],ans;
int n,m,num;
int cmp(const P &a,const P &b){
    return a.h<b.h;
}
void pushUp(ll rt,ll l,ll r){
    if(t[rt].c)t[rt].sum=x[r+1]-x[l];//r+1是因为节点[l,r]表示区间[x[l],x[r+1]]。
    else if(l==r)t[rt].sum=0;
    else t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
}
void update(ll s,ll e,ll rt,ll l,ll r,ll v){
    if(s<=l&&r<=e) t[rt].c+=v;
    else {
        if(l>e||r<s)return;
        ll m=l+r>>1;
        update(s,e,rt<<1,l,m,v);
        update(s,e,rt<<1|1,m+1,r,v);
    }
    pushUp(rt,l,r);
}
int main()
{
    while(scanf("%d",&n),n){
        int k=0;
        for(int i=0;i<n;i++){
            scanf("%lf%lf%lf%lf",&a,&b,&c,&d);
            x[++k]=a,p[k]=(P){a,c,b,1};
            x[++k]=c,p[k]=(P){a,c,d,-1};
        }
        sort(x+1,x+1+k);
        sort(p+1,p+1+k,cmp);
        m=1;
        for(int i=1;i<=k;i++)
            if(x[i]!=x[i-1])x[m++]=x[i];
        ans=0;
        for(int i=1;i<=k;i++){//共k条线段,每次计算p[i].h到p[i+1].h之间的面积，第k次相当于清空所有，酱就不用初始化线段树了。
            int l=lower_bound(x,x+m,p[i].s)-x;
            int r=lower_bound(x,x+m,p[i].e)-x-1;//r-1同上原因
            update(l,r,1,0,m-1,p[i].f);
            ans+=t[1].sum*(p[i+1].h-p[i].h);
        }
        printf("Test case #%d\nTotal explored area: %.2lf\n\n",++num,ans);
    }
    return 0;
}

#include<cstdio>
#include<algorithm>
#define dd double
using namespace std;
#define N 201
struct P{dd x,y1,y2;int f;}p[N];
struct TREE{dd y1,y2,x;int c,f;}tree[N<<5];
dd x1,y1,x2,y2,y[N];
int n,k,num;
int cmp(const P &a,const P &b){
    return a.x<b.x;
}
void build(int i,int l,int r){
    tree[i].c=tree[i].f=0;
    tree[i].y1=y[l];//直接将线段树节点代表的区间存在线段树里
    tree[i].y2=y[r];
    if(l+1==r){
        tree[i].f=1;
        return;
    }
    int mk=(l+r)>>1;
    build(i<<1,l,mk);
    build(i<<1|1,mk,r);
}
dd insert(int i,dd x,dd l,dd r,int flag){
    if(r<=tree[i].y1||l>=tree[i].y2)
        return 0;
    if(tree[i].f){//离散后的一个最小区间，叶子节点
        dd ans;
        if(tree[i].c>0)//全覆盖
            ans=(x-tree[i].x)*(tree[i].y2-tree[i].y1);//(当前x-该区间最后的x)*区间高度
        else
            ans=0;
        tree[i].x=x;//该区间最新的x
        tree[i].c+=flag;//更新覆盖
        return ans;
    }
    return insert(i<<1,x,l,r,flag)+insert(i<<1|1,x,l,r,flag);
}
int main(){
    while(scanf("%d",&n),n){
        k=0;
        for(int i=1;i<=n;i++){
            scanf("%lf%lf%lf%lf",&x1,&y1,&x2,&y2);
            y[++k]=y1,p[k]=(P){x1,y1,y2,1};
            y[++k]=y2,p[k]=(P){x2,y1,y2,-1};
        }
        sort(y+1,y+k+1);
        sort(p+1,p+k+1,cmp);
        //没有去重，其实数量少，没必要。
        build(1,1,k);
        dd ans=0;
        for(int i=1;i<=k;i++)
            ans+=insert(1,p[i].x,p[i].y1,p[i].y2,p[i].f);
        printf("Test case #%d\nTotal explored area: %.2f\n\n",++num,ans);
    }
    return 0;
}