HDU - 1255 覆盖的面积

题目链接点这里

这道题，，，很久以前做过，，当时写了个暴力，居然过了，，然后一直想写下线段树的，，，结果一直被各种事搞得忘了。。。现在终于填坑了。。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define mem(x,y) memset(x,y,sizeof(x))
#define FIN freopen("input.txt","r",stdin)
#define fuck(x) cout<<x<<endl
const int MX=111111;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
double eps=0.00000001;
double sum[MX<<2][2];
int cnt[MX<<2];
int n;
double arr[2*MX];
struct Seg
{
    double x1,x2,h;
    int flag;
    Seg() {}
    Seg(double x1,double x2,double h,int flag):x1(x1),x2(x2),h(h),flag(flag) {}
    bool operator <(const Seg &a )const
    {
        return h<a.h;
    }

} seg[2*MX];
int find(double x,double *a,int l,int r)
{
    int m;
    while(l<r)
    {
        m=(l+r)>>1;
        if(a[m]==x) return m;
        else if(a[m]<x) l=m+1;
        else r=m-1;
    }
    return l;
}
void pushup(int rt,int l,int r)
{
    if(l==r) if(cnt[rt]==1)sum[rt][0]=arr[r]-arr[l-1],sum[rt][1]=0;
        else if(cnt[rt]>=2)sum[rt][1]=arr[r]-arr[l-1],sum[rt][0]=0;
        else sum[rt][1]=sum[rt][0]=0;
    else  if(cnt[rt]==1)
    {
        sum[rt][1]=sum[rt<<1][0]+sum[rt<<1|1][0]+sum[rt<<1][1]+sum[rt<<1|1][1];
        sum[rt][0]=arr[r]-arr[l-1]-sum[rt][1];
    }
    else if(cnt[rt]>=2)
    {
        sum[rt][1]=arr[r]-arr[l-1];
        sum[rt][0]=0;
    }

    else
    {
        sum[rt][1]=sum[rt<<1][1]+sum[rt<<1|1][1];
        sum[rt][0]=sum[rt<<1][0]+sum[rt<<1|1][0];
    }
}
void update(int d,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        cnt[rt]+=d;
        pushup(rt,l,r);
        return ;
    }
    int m=(l+r)>>1;
    if(L<=m) update(d,L,R,lson);
    if(m+1<=R) update(d,L,R,rson);
    pushup(rt,l,r);
}
int main()
{
    FIN;
    int T;
    cin>>T;
    while(T--)
    {
        scanf("%d",&n);
        int k=0;
        for(int i=0; i<n; i++)
        {
            double x1,y1,x2,y2;
            scanf("%lf%lf%lf%lf",&x1,&y1,&x2,&y2);
            seg[2*i]=Seg(x1,x2,y1,1);
            seg[2*i+1]=Seg(x1,x2,y2,-1);
            arr[2*i]=x1;
            arr[2*i+1]=x2;
        }
        sort(seg,seg+2*n);
        sort(arr,arr+2*n);
        k=unique(arr,arr+2*n)-arr;
        double ans=0;
        mem(cnt,0);
        mem(sum,0);
        for(int i=0; i<2*n; i++)
        {
            int l=find(seg[i].x1,arr,0,k-1);
            int r=find(seg[i].x2,arr,0,k-1);
            if(i)ans+=sum[1][1]*(seg[i].h-seg[i-1].h);
            update(seg[i].flag,l+1,r,0,k-1,1);
        }
        printf("%.2f\n",ans);
    }
    return 0;
}