線段樹+掃描線(有關掃描線的理解)
掃描線:
下面是來自soar轉載的一篇部落格。
這篇部落格解決了我對算區間長度時的不理解。實際上這個線段樹的葉子節點儲存的是這個點x座標到下一個x座標(排序後的)的區間長度。
題意:
二維平面有n個平行於座標軸的矩形,現在要求出這些矩形的總面積. 重疊部分只能算一次.
分析:
線段樹的典型掃描線用法.
首先假設有下圖兩個矩陣,我們如果用掃描線的方法如何計算它們的總面積呢?
首先我們將矩形的上下邊分為上位邊(即y座標大的那條平行於x軸的邊),和下位邊(y座標小的平行於x軸的邊).然後我們把所有矩形的上下位邊按照他們y座標從小到大排序,可以得到4條掃描線:
又因為上面2個矩形有4個不同的浮點數x座標,所以我們需要把x座標離散化,這樣才能用線段樹來維護資訊.所以我們這樣離散化:
由上圖可知,4個不同的x座標把x軸分成了3段有效的區間.這裡要注意我們線段樹中每個葉節點(控制區間[L,L])不是指X[L]座標,而是指區間[X[L],X[L+1]].線段樹中其他節點控制的區間[L,R],也是指的x座標軸的第L個區間到第R個區間的範圍,也就是X[L]到X[R+1]座標的範圍.
然後我們Y座標從小到大的順序讀入每條掃描線,並維護當前我們所讀入的所有掃描線能有效覆蓋X軸的最大長度sum[1].這裡特別要注意如果我們讀入的掃描線是矩形的下位邊,那麼我們就使得該範圍的標記cnt位+1,如果是上位邊,那麼該範圍的cnt就-1.所以如果cnt=0時,表示該節點控制的範圍沒有被覆蓋,只要cnt!=0 就表示該節點控制的幾塊區間仍然被覆蓋.
下面依次讀入每條矩陣邊,來一一分析,首先是讀入第一條矩陣邊:
我們讀入了矩形1的下位邊,那麼該區域的cnt就+1=1了,所以該區域[10,20]就被覆蓋了,然後可以推出整個區域被覆蓋的長度是10.再根據第二條掃描線離第一條掃描線的高度差為5.所以不管你第二條掃描線是哪個矩形的什麼邊,或者能覆蓋到X軸的什麼範圍,我上圖中藍色的矩形面積肯定是要算到總面積裡面去的.即總面積ret+=sum[1]*(掃描線2的高度-掃描線1的高度). (想想看是不是這樣).
下面讀第二條掃描線:
由於第二條掃描線也是下位邊,所以[15,20]和[20,25]的cnt+1.使得我們覆蓋的範圍變成了[10,25]了,並且第3條掃描線在20高度,所以這次我們必然增加的面積是上面深藍色的長條=sum[1]*(掃描線3的高度-掃描線2的高度).
下面我們要讀第三條掃描線了:
由於第三條掃描線是區間[10,20]的上位邊,所以對應區間的cnt要-1,所以使得區間[10,15]的cnt=0了,而[15,20]區間的cnt-1之後變成了1.[20,25]的cnt仍然為1,不變.所以當前覆蓋的有效x軸長度為10,即區間[15,25].所以增加的面積是圖中褐色的部分.
到此,矩形的面積和就算出來了.由於對於任一矩形都是先讀下位邊(cnt+1),再讀上位邊(cnt-1)的,所以在更新線段樹的過程中,任意節點的cnt都是>=0的.
下面說程式碼實現部分:
首先建立一個node結構體用來儲存每條掃描線,node中有資訊:
l: 表示掃描線的左端x座標
r:表示掃描線的右端x座標
h: 表示掃描線的高度
d: 為1或-1,標記掃描線是矩形的上位還是下位邊.
我們首先讀入所有矩形的資訊,每讀入一個矩形資訊我們就更新兩條掃描線,並且把矩形的兩個端點x座標放入X[MAXN]陣列中,
然後我們對node和X都排序,node按h值從小到大排序.
X按從小到大排序.
然後我們在X的本地陣列內,對X去重,並且用k表示一共有多少個X.
當我們需要找到第i個區域的兩端點座標時,只需要X[i]和X]i+1].
線段樹維護cnt(根本資訊)和sum兩個資訊,其中sum為double,cnt為int型.
cnt: >=0時表示本節點控制的區域內下位邊個數-上位邊個數的結果.如果==-1時,表示本節點左右子樹的上下位邊數不一致.
sum: 本節點控制的區域內cnt值不為0的區域總長度.
線段樹操作:
PushDown(i,l,r):如果cnt!=-1,那麼下放cnt資訊,並更新子節點的sum資訊.
PushUp(i,l,r): 根據子節點的cnt值和sum值更新父節點的cnt和sum值.
update(ql,qr,v,i,l,r): 使得[ql,qr]與[l,r]區間的公共部分cnt值+v.
如果ql<=l && r<=qr 且 cnt[i]!=-1的話,直接更新並return
否則先PushDown
在一次遞迴更新左右兒子
最後PushUp.
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;
const int MAX=200+10;
int mark[MAX<<2];//記錄某個區間的下底邊個數
double sum[MAX<<2];//記錄某個區間的下底邊總長度
double hash[MAX];//對x進行離散化,否則x為浮點數且很大無法進行線段樹
//以橫座標作為線段(區間),對橫座標線段進行掃描
//掃描的作用是每次更新下底邊總長度和下底邊個數,增加新面積
struct seg{//線段
double l,r,h;
int d;
seg(){}
seg(double x1,double x2,double H,int c):l(x1),r(x2),h(H),d(c){}
bool operator<(const seg &a)const{
return h<a.h;
}
}s[MAX];
void Upfather(int n,int left,int right){
if(mark[n])sum[n]=hash[right+1]-hash[left];//表示該區間整個線段長度可以作為底邊
else if(left == right)sum[n]=0;//葉子結點則底邊長度為0(區間內線段長度為0)
else sum[n]=sum[n<<1]+sum[n<<1|1];
}
void Update(int L,int R,int d,int n,int left,int right){
if(L<=left && right<=R){//該區間是當前掃描線段的一部分,則該區間下底邊總長以及上下底邊個數差更新
mark[n]+=d;//更新底邊相差差個數
Upfather(n,left,right);//更新底邊長
return;
}
int mid=left+right>>1;
if(L<=mid)Update(L,R,d,n<<1,left,mid);
if(R>mid)Update(L,R,d,n<<1|1,mid+1,right);
Upfather(n,left,right);
}
int search(double key,double* x,int n){
int left=0,right=n-1;
while(left<=right){
int mid=left+right>>1;
if(x[mid] == key)return mid;
if(x[mid]>key)right=mid-1;
else left=mid+1;
}
return -1;
}
int main(){
int n,num=0;
double x1,x2,y1,y2;
while(cin>>n,n){
int k=0;
for(int i=0;i<n;++i){
cin>>x1>>y1>>x2>>y2;
hash[k]=x1;
s[k++]=seg(x1,x2,y1,1);
hash[k]=x2;
s[k++]=seg(x1,x2,y2,-1);
}
sort(hash,hash+k);
sort(s,s+k);
int m=1;
for(int i=1;i<k;++i)//去重複端點
if(hash[i] != hash[i-1])hash[m++]=hash[i];
double ans=0;
//memset(mark,0,sizeof mark);
//memset(sum,0,sizeof sum);如果下面是i<k-1則要初始化,因為如果對第k-1條線段掃描時會使得mark,sum為0才不用初始化的
for(int i=0;i<k;++i){//掃描線段
int L=search(s[i].l,hash,m);
int R=search(s[i].r,hash,m)-1;
Update(L,R,s[i].d,1,0,m-1);//掃描線段時更新底邊長度和底邊相差個數
ans+=sum[1]*(s[i+1].h-s[i].h);//新增加面積
}
printf("Test case #%d\nTotal explored area: %.2lf\n\n",++num,ans);
}
return 0;
}
/*
這裡注意下掃描線段時r-1:int R=search(s[i].l,hash,m)-1;
計算底邊長時r+1:if(mark[n])sum[n]=hash[right+1]-hash[left];
解釋:假設現在有一個線段左端點是l=0,右端點是r=m-1
則我們去更新的時候,會算到sum[1]=hash[mid]-hash[left]+hash[right]-hash[mid+1]
這樣的到的底邊長sum是錯誤的,why?因為少算了mid~mid+1的距離,由於我們這利用了
離散化且區間表示線段,所以mid~mid+1之間是有長度的,比如hash[3]=1.2,hash[4]=5.6,mid=3
所以這裡用r-1,r+1就很好理解了
*/
無註釋精簡版:
#include<bits/stdc++.h>
using namespace std;
#define lson i<<1,l,m
#define rson i<<1|1,m+1,r
const int maxn=222;
double x[maxn];
struct node
{
double l,r,h;//左右座標,高度
int d;//標記上位邊還是下位邊
node() {}
node(double l,double r,double h,int d):l(l),r(r),h(h),d(d) {}
bool operator < (const node &a)const
{
return h<a.h;
}
} line[maxn];
int cnt[maxn<<2];
double sum[maxn<<2];
void pushup(int i,int l,int r)
{
if(cnt[i])
{
sum[i]=x[r+1]-x[l];
}
else
{
sum[i]=sum[i<<1]+sum[i<<1|1];
}
}
void update(int ql,int qr,int v,int i,int l,int r)
{
if(ql<=l && qr>=r)
{
cnt[i]+=v;
pushup(i,l,r);
return ;
}
int m=(l+r)>>1;
if(ql<=m)update(ql,qr,v,lson);
if(qr>m)update(ql,qr,v,rson);
pushup(i,l,r);
}
int main()
{
int q;
int kase=0;
while(cin>>q&&q)
{
memset(cnt,0,sizeof(cnt));//相當於build
memset(sum,0,sizeof(sum));//相當於build
int n=0,m=0;
for(int i=1; i<=q; i++)
{
double x1,y1,x2,y2;
scanf("%lf%lf%lf%lf",&x1,&y1,&x2,&y2);
x[++n]=x1;
x[++n]=x2;
line[++m]=node(x1,x2,y1,1);
line[++m]=node(x1,x2,y2,-1);
}
sort(x+1,x+1+n);
sort(line+1,line+1+m);
int k=1;
/* for(int i=2;i<=n;i++)//去重
{
if(x[i]!=x[i-1])x[++k]=x[i];
}*/
k=unique(x+1,x+n+1)-x-1;//直接用STL中的unique函式。
double ans=0.0;
for(int i=1; i<m; i++)
{
int l=lower_bound(x+1,x+k+1,line[i].l)-x;
int r=lower_bound(x+1,x+k+1,line[i].r)-x;
r--;
if(l<=r)update(l,r,line[i].d,1,1,k-1);
ans+=sum[1]*(line[i+1].h-line[i].h);
}
printf("Test case #%d\nTotal explored area: %.2f\n\n",++kase,ans);
}
}