題意:
給定n個矩形(0 <= n < 5000)的左下角座標(x1,y1)和右上角座標(x2,y2) (-10000 <= x1,x2,y1,y2 <= 10000)
求所有矩形重合後的圖形的周長,如下圖(圖片來自POJ 1177):
做法:線段樹掃描線。
由於值域不大,所以不需要離散化,直接將Y值向正方向平移10001個單位,然後用線段樹直接做。
掃描線就是用垂直於x軸的線來從左到右掃描整個圖形,然後用線段樹來記錄這條線上有多少個單位被覆蓋。
struct Node{
int cover;//完全覆蓋層數
int lines;//分成多少個線段
bool L,R;//左右端點是否被覆蓋
int CoverLength;//覆蓋長度
int Length;//總長度
};
爲了計算與x軸平行的邊,需要記錄掃描線上有多少條線段,用lines 表示,然後 L,R表示該區間的左右端點是否被覆蓋,方便合併區間的時候維護 lines。
Length是本區間的總長度,CoverLength是本區間被覆蓋的長度,cover是指本區間被完全覆蓋的次數,非完全覆蓋不改變cover值。
然後,每個矩形,遇到左側的邊就將這條邊加入掃描線,遇到右側的邊就把這條邊從掃描線刪除。
每次操作後更新周長。
具體見代碼:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define inf 0x1fffffff
#define out(i) <<#i<<"="<<(i)<<" "
#define OUT1(a1) cout out(a1) <<endl
#define OUT2(a1,a2) cout out(a1) out(a2) <<endl
#define OUT3(a1,a2,a3) cout out(a1) out(a2) out(a3)<<endl
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define maxn 20007 //直接按y值建樹
using namespace std;
struct Node{
int cover;//完全覆蓋層數
int lines;//分成多少個線段
bool L,R;//左右端點是否被覆蓋
int CoverLength;//覆蓋長度
int Length;//總長度
Node(){}
Node(int cover,int lines,bool L,bool R,int CoverLength):cover(cover),lines(lines),L(L),R(R),CoverLength(CoverLength){}
Node operator +(const Node &B){//連續區間的合併
Node C;
C.cover=0;
C.lines=lines+B.lines-(R&&B.L);
C.CoverLength=CoverLength+B.CoverLength;
C.L=L;C.R=B.R;
C.Length=Length+B.Length;
return C;
}
void Show(){
printf("cover=%d lines=%d L=%d R=%d CoverLength=%d Length=%d\n",cover,lines,L,R,CoverLength,Length);
}
}K[maxn<<2];
void PushUp(int rt){//更新非葉節點
if(K[rt].cover){
K[rt].CoverLength=K[rt].Length;
K[rt].L=K[rt].R=K[rt].lines=1;
}
else{
K[rt]=K[rt<<1]+K[rt<<1|1];
}
}
void Build(int l,int r,int rt){//建樹
if(l==r){
K[rt].L=K[rt].R=K[rt].cover=K[rt].lines=K[rt].CoverLength=0;
K[rt].Length=1;
return;
}
int m=(l+r)>>1;
Build(ls);
Build(rs);
K[rt].cover=0;
PushUp(rt);
}
void Cover(int L,int R,int l,int r,int rt){//覆蓋
if(L <= l && r <= R){
if(!K[rt].cover){//如果覆蓋前爲空,則更新節點內容
K[rt].L=K[rt].R=true;
K[rt].lines=1;
K[rt].CoverLength=K[rt].Length;
}
K[rt].cover++;
return;
}
int m=(l+r)>>1;
if(L <= m) Cover(L,R,ls);
if(R > m) Cover(L,R,rs);
PushUp(rt);
}
void Uncover(int L,int R,int l,int r,int rt){//移除覆蓋
if(L <= l && r <= R){//如果移除後爲空,則更新節點
--K[rt].cover;
if(!K[rt].cover) {
if(l==r){//特判葉節點
K[rt].L=K[rt].R=K[rt].cover=K[rt].lines=K[rt].CoverLength=0;
}
else K[rt]=K[rt<<1]+K[rt<<1|1];//非葉節點用重載的+號
}
return;
}
int m=(l+r)>>1;
if(L <= m) Uncover(L,R,ls);
if(R > m) Uncover(L,R,rs);
PushUp(rt);
}
int n;
struct Lines{//線段
int x,y1,y2,c;//x:線的橫座標 y1,y2:線的縱座標端點 c:加入1 移除0
Lines(){}
Lines(int x,int y1,int y2,int c):x(x),y1(y1),y2(y2),c(c){}
bool operator < (const Lines &B)const{return x < B.x;} //重載小於號方便排序
}L[10000];
int main(void)
{
while(~scanf("%d",&n)){
int x1,y1,x2,y2;
for(int i=0;i<n;++i){
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
L[i<<1]=Lines(x1,y1+10001,y2+10001,1);
L[i<<1|1]=Lines(x2,y1+10001,y2+10001,0);
}
int nL=n<<1;
sort(L,L+nL);//按x值排序
int PreX=L[0].x;//前X座標
int ANS=0;//目前累計答案
int PreLength=0;//前線段總長
int PreLines=0;//前線段數量
Build(1,20001,1);
for(int i=0;i<nL;++i){
//操作
if(L[i].c) Cover(L[i].y1,L[i].y2-1,1,20001,1);
else Uncover(L[i].y1,L[i].y2-1,1,20001,1);
//更新橫向的邊界
ANS+=2*PreLines*(L[i].x-PreX);
PreLines=K[1].lines;
PreX=L[i].x;
//更新縱向邊界
ANS+=abs(K[1].CoverLength-PreLength);
PreLength=K[1].CoverLength;
}
//輸出答案
printf("%d\n",ANS);
}
return 0;
}