POJ 1177 Picture (線段樹掃描線)

題意:

給定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;
}









發佈了96 篇原創文章 · 獲贊 375 · 訪問量 17萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章