扫描线模板题
题目大意:
给若干矩形,求组成图形的总面积
思路:线段树扫描线
用到的小算法:离散化+线段树
不细讲了,主要记录一下学习过程中的细节和注意点(花了四天勉强搞懂了扫描线T - T)
为什么不用懒标记呢?
我的理解是:根据矩形的对称性,两宽两长,所以加入线段树的扫描线一定都是成对,最后都是一一抵消的(不知道说的准不准确),只用向根维护总区间的有效长度即可,不用查询根一下的结点
为什么看到代码xx[r+1] -xx[l]有个+1
看到两种建树写法,首先明确因为这里对所有区间进行了离散化,再存在树上。与普通线段树不同的是,树上每个点代表的是一个区间,如果一共n个横坐标,树根节点维护的区间就是[1,n-1],因为一共n-1个“小片段”——每个区间只包括两个横坐标。r+1实际是对离散化的区间进行“还原”操作,存的时候将区间[l,r]压成[l,r-1]存到树上,要用到区间的时候自然要回归它的真实区间啦~
两种建树操作
写法一:
利用上面的思想,树上每个点代表的是一个区间,这种是最常见的写法(代码在最后,有详注)
写法二:
少见的写法,贴个建树和更新代码。
树上每个结点维护的是实际区间,建树时[100,200]如何分成两段呢,如果需要分成[100,120]和[120,200],普通线段树会不断对[100,200]“二分”,到叶子结点为止。
这里也是离散的存实际区间,直接分成[100,120]和[120,200],所以建右子树时是build ( rson , mid , r ,而不是mid+1 。
另外这样建树是离散的区间存区间比如[1,2]代表[100,200],和上面点代表区间有区别,所以不能出现树结点l==r的情况(两个端点不算区间呢),所以到r-l==1时return一下
注意点
去重操作可做可不做,没影响
避免将变量名写为y1 , cmath里y1的常量,会出现冲突,当然万能库也是
void build(int p, int l, int r)//这里的l,r是离散化后的区间
{
tree[p].l = xx[l];
tree[p].r = xx[r];
if(r-l<=1) return;//特殊线段树,不用端点
int mid = (l+r)>>1;
build(ls , l , mid);
build(rs , mid , r);
//建树无需维护值
}
void update(int p, int x, int y, int k)//k为±1
{
int l = tree[p].l;//实际区间
int r = tree[p].r;
if( x<=l && r<= y)
{
tree[p].cover += k;
pushup(p);//pushup维护
return;
}
//这里和普通线段树不一样
if( x<tree[ls].r) update(ls,x,y,k);//与左儿子的右端点比较
if( y>tree[rs].l) update(rs,x,y,k);//与右儿子的左端点比较
pushup(p);//维护
}
//线段树扫描线,矩形并
#include <iostream>
#include <cstring>
//#include <cmath>//cmath里有y1常量,避免混用
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <map>
#include <vector>
#include <queue>
using namespace std;
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define mst(v,s) memset(v,s,sizeof(v))
#define pb push_back
#define IOS ios::sync_with_stdio(false)
#define int long long
#define all(v) v.begin(),v.end()
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
#define ls p<<1
#define rs p<<1|1
typedef long long ll;
const int N=2e6+10;
int n;
struct L
{
int l,r;//扫描线左右端点
int h;//高度
int f;//1为下边 -1位上边
bool operator < (const L & b) const//按照扫描线从低到高排序
{
return h < b.h;
}
}line[N];
struct node//维护树上信息,注意树上每个点代表的是一个区间!!!
{
int l,r;//离散化后的左右端点
int len;//维护区间被覆盖的长度
int cover;//维护覆盖的次数
}tree[N];
int xx[N];//存所有横坐标
void pushup(int p)//合并
{
int l = tree[p].l,r = tree[p].r;
if( tree[p].cover ) tree[p].len = xx[r+1] - xx[l];
else tree[p].len = tree[ls].len + tree[rs].len;
}
void build(int p ,int l, int r)//建树,每个点代表一个区间
{
tree[p].l = l;//离散地建树 eg.[1,2]其实是两个区间 eg[1,1]其实是一个区间
tree[p].r = r;
if( l==r ) return ;
int mid=(l+r)>>1;
build(ls,l,mid);
build(rs,mid+1,r);
}
void update(int p, int l ,int r ,int f)//更新区间,l,r是目标区间离散化后的区间
{
int L = tree[p].l;
int R = tree[p].r;
if( l <= L && R<=r )//离散化后的目标区间和线段树区间(建树的时候就是离散化的)
{
tree[p].cover+=f;//更新覆盖次数
pushup(p);
return;
}
int mid=(L+R)>>1;
if( l<=mid ) update(ls,l,r,f);//更新左子树
if( r>=mid+1 ) update(rs,l,r,f);//更新右子树
pushup(p);//合并子树,向上维护
}
signed main()
{
///!!!
// freopen("data.txt","r",stdin);
///!!!
IOS;
cin>>n;
_for(i,1,n)
{
int x1,y1,x2,y2;
cin>>x1>>y1>>x2>>y2;
xx[i] = x1,xx[i+n] = x2;//离散化读所有横坐标
line[i] = (L){x1,x2,y1,1};//无所谓顺序,反正要排序
line[i+n] = (L){x1,x2,y2,-1};
}
//扫描线和横坐标排序
sort(xx+1,xx+1+2*n);
sort(line+1,line+1+2*n);
int tot = unique(xx+1,xx+1+2*n)-xx-1;//去重,不去也行的
build(1,1,tot-1);//区间对应端点,一共tot个横坐标,tot-1个“片段”,建树每个点代表区间,所以是tot-1
ll ans=0;
_for(i,1,2*n)
{
int xl = lower_bound(xx+1,xx+1+tot,line[i].l) - xx;//离散找左端点
int xr = lower_bound(xx+1,xx+1+tot,line[i].r) - xx-1;//离散找右端点
ans += tree[1].len *(line[i].h - line[i-1].h );
update(1,xl,xr,line[i].f);//将扫描线插入到线段树
}
cout<<ans<<endl;
}