POJ 1177 Picture（矩形并的周长）

题目链接：
POJ 1177 Picture

//926K 16MS#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <algorithm>#include <climits>#define lson(x) (x<<1)#define rson(x) ((x<<1)|1)using namespace std;const int maxn=5050;int n,x1,x2,yy1,y2;int xx[maxn<<2];struct SegTree{    int left,right,len,num,flag;//len是有效长度，num是有几条线段,flag记录状态    bool lcover,rcover;//记录区间左右端点是否存在竖边}segtree[maxn<<4];struct Line{    int x1,x2,y,flag;    bool operator < (const Line a) const{        return y<a.y;    }}line[maxn<<2];void build(int left,int right,int cur){    segtree[cur].left=left;    segtree[cur].right=right;    segtree[cur].len=segtree[cur].num=segtree[cur].flag=0;    segtree[cur].lcover=segtree[cur].rcover=false;    if(left+1==right) return ;    int mid=(left+right)>>1;    build(left,mid,lson(cur));    build(mid,right,rson(cur));}void calc_len(int cur){    int left=segtree[cur].left;    int right=segtree[cur].right;    if(segtree[cur].flag>0){        segtree[cur].len=xx[right]-xx[left];        segtree[cur].num=1;        segtree[cur].lcover=segtree[cur].rcover=true;    }else if(left+1==right){        segtree[cur].len=segtree[cur].num=0;        segtree[cur].lcover=segtree[cur].rcover=false;    }else{        segtree[cur].len=segtree[lson(cur)].len+segtree[rson(cur)].len;        segtree[cur].num=segtree[lson(cur)].num+segtree[rson(cur)].num;        segtree[cur].lcover=segtree[lson(cur)].lcover;        segtree[cur].rcover=segtree[rson(cur)].rcover;        if(segtree[lson(cur)].rcover&&segtree[rson(cur)].lcover)//左右儿子的线段可以衔接            segtree[cur].num--;//合并成一个线段，所以线段数量-1    }    return;}void update(int a,int b,int flag,int cur){    int left=segtree[cur].left;    int right=segtree[cur].right;    if(left==a&&right==b){        segtree[cur].flag+=flag;        calc_len(cur);        return;    }    int mid=(left+right)>>1;    if(b<=mid) update(a,b,flag,lson(cur));    else if(a>=mid) update(a,b,flag,rson(cur));    else{        update(a,mid,flag,lson(cur));        update(mid,b,flag,rson(cur));    }    calc_len(cur);}int main(){#ifdef  LOCAL    freopen("in.txt","r",stdin);#endif    while(~scanf("%d",&n)){        int tot=0;        for(int i=0;i<n;i++){            scanf("%d %d %d %d",&x1,&yy1,&x2,&y2);            line[tot].x1=x1,line[tot].x2=x2,line[tot].y=yy1,line[tot].flag=1;            line[tot+1].x1=x1,line[tot+1].x2=x2,line[tot+1].y=y2,line[tot+1].flag=-1;            xx[tot]=x1,xx[tot+1]=x2;            tot+=2;        }        sort(xx,xx+tot);        int m=unique(xx,xx+tot)-xx;        build(0,m-1,1);        sort(line,line+tot);        int ans=0,last=0,a,b;        for(int i=0;i<tot-1;i++){            a=lower_bound(xx,xx+m,line[i].x1)-xx;            b=lower_bound(xx,xx+m,line[i].x2)-xx;            update(a,b,line[i].flag,1);            ans+=(line[i+1].y-line[i].y)*segtree[1].num*2;//竖边数量是线段数量的两倍            ans+=abs(segtree[1].len-last);            //因为加上边时flag为-1，这时线段树中这条边就不存在了，但是求周长时是应该加上的            //而且在上一棵线段树中是有的，所以要用abs            last=segtree[1].len;        }        a=lower_bound(xx,xx+m,line[tot-1].x1)-xx;        b=lower_bound(xx,xx+m,line[tot-1].x2)-xx;        update(a,b,line[tot-1].flag,1);        ans+=abs(segtree[1].len-last);        printf("%d\n",ans);    }    return 0;}