蓝书（算法竞赛进阶指南）刷题记录——POJ1177 Picture（扫描线+线段树）

题目：POJ1177.
题目大意：给定$n$个矩形的左下角$(x{1}_i,y{1}_i)$和右上角$(x{2}_i,y{2}_i)$，求它们并的周长.
$0\le n<5000,x{1}_i,x{2}_i,y{1}_i,y{2}_i\in [-10^4,10^4]\cap Z$.

矩形周长并的板子题.

首先考虑把$n$个矩形拆成$2n$条与$y$轴平行（与$x$轴平行也没关系）的扫描线，前面的线权值为$1$，后面的线权值为$-1$，这个时候我们分开考虑与$x$轴平行的线和与$y$轴平行的线对周长的贡献.

先来考虑与$y$轴平行的线.发现只有当一段的值要从$0$变成$1$或从$1$变成$0$时才会有对应长度的贡献，其实这个值就是这一次被覆盖的线段总长减去下一次被覆盖的线段总长的绝对值.

再来考虑与$x$轴平行的线.发现每个时刻每一段极大的非$0$段都会有两份的贡献，用线段树维护一下也是可以实现的.

时间复杂度$O(n\log n)$.

代码如下：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
  using namespace std;

#define Abigail inline void
typedef long long LL;

const int N=5000;

struct seg{
  int x,l,r,v;
  seg(int X=0,int L=0,int R=0,int V=0){x=X;l=L;r=R;v=V;}
  bool operator < (const seg &p)const{return x<p.x;}
}e[N*2+9];
int ord[N*2+9],n,ans;

int lower(int k){
  int l=1,r=n<<1,mid=l+r>>1;
  for (;l<r;mid=l+r>>1)
    k>ord[mid]?l=mid+1:r=mid;
  return l;
}

struct tree{
  int v,tag,lc,rc,sum,cnt;
  //v表示这一段的实际长度
  //tag为这一段被整体覆盖的数量
  //lc,rc分别表示左右端点是否被覆盖
  //sum表示未被覆盖的段的长度之和
  //cnt表示线段数量
}tr[N*16+9];

void Pushup(int k){
  if (tr[k].tag){
  	tr[k].sum=tr[k].v;
  	tr[k].lc=tr[k].rc=1;
  	tr[k].cnt=1;
    return;
  }
  int ls=k<<1,rs=k<<1|1;
  tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;
  tr[k].lc=tr[k<<1].lc;tr[k].rc=tr[k<<1|1].rc;
  tr[k].cnt=tr[k<<1].cnt+tr[k<<1|1].cnt-(tr[k<<1].rc&&tr[k<<1|1].lc?1:0); 
}

void Build(int L,int R,int k=1){
  tr[k]=tree();
  if (L==R) {tr[k].v=ord[L+1]-ord[L];tr[k<<1]=tr[k<<1|1]=tree();return;}
  int mid=L+R>>1;
  Build(L,mid,k<<1),Build(mid+1,R,k<<1|1);
  tr[k].v=tr[k<<1].v+tr[k<<1|1].v;
}

void Change_add(int L,int R,int v,int l,int r,int k=1){
  if (L==l&&R==r){tr[k].tag+=v;Pushup(k);return;}
  int mid=l+r>>1;
  if (R<=mid) Change_add(L,R,v,l,mid,k<<1);
  else if (L>mid) Change_add(L,R,v,mid+1,r,k<<1|1);
    else Change_add(L,mid,v,l,mid,k<<1),Change_add(mid+1,R,v,mid+1,r,k<<1|1);
  Pushup(k);
}

Abigail into(){
  scanf("%d",&n);
  int a1,a2,b1,b2;
  for (int i=1;i<=n;++i){
  	scanf("%d%d%d%d",&a1,&b1,&a2,&b2);
  	e[i*2-1]=seg(a1,b1,b2,1);e[i*2]=seg(a2,b1,b2,-1);
  	ord[i*2-1]=b1;ord[i*2]=b2;
  }
}

Abigail work(){
  sort(e+1,e+1+2*n);
  sort(ord+1,ord+1+2*n);
  Build(1,n<<1);
  for (int i=1;i<=n<<1;++i)
    e[i].l=lower(e[i].l),e[i].r=lower(e[i].r);
  int last=0;
  for (int i=1;i<=n<<1;++i){
    Change_add(e[i].l,e[i].r-1,e[i].v,1,n<<1);
  	ans+=abs(tr[1].sum-last);last=tr[1].sum;
    ans+=tr[1].cnt*(e[i+1].x-e[i].x)<<1;
  }
}

Abigail outo(){
  printf("%d\n",ans);
}

int main(){
  into();
  work();
  outo();
  return 0;
}