没有传送门。
Description
因为对polo忍无可忍, dzf使用圣剑在地上划出了许多纵横交错的沟壑来泄愤。这些沟壑都严格与X轴平行或垂直。
polo嘲笑了dzf无聊的行为,然后做了一件更加无聊的事。他蹲下来数这些沟壑的条数。数着数着,polo意识到一个问题,那就是因为圣剑的威力太大,划出的沟壑太多,地面就会塌陷。而如果两条水平的沟壑和两条垂直的沟壑相交组成了一个矩形,那么塌陷的危险就会进一步增加。现在polo已经数了n条沟壑,他想知道这些沟壑组成了多少个矩形。
Input
第一行一个数n,接下来每行4个数x1,y1,x2,y2,表示沟壑的两个端点(x1,y1),(x2,y2)
Output
一个数,组成的矩形个数。
Sample Input
#1
4
0 0 1 0
0 0 0 1
1 1 1 -1
1 1 0 1
#2
8
1 0 4 0
2 1 2 0
0 0 0 3
2 2 2 3
3 3 3 -1
0 3 4 3
4 1 -1 1
3 2 -1 2
Sample Output
#1
1
#2
6
Hint
【数据说明】
对于30%的数据,1<=n<=100
对于60%的数据,1<=n<=600
对于100%的数据,1<=n<=2000,坐标绝对值小于2^31,任意两条与X轴水平的沟壑之间没有交点,任意两条与X轴垂直的沟壑没有交点。
Solution
- 容易想到一个O(n^3)暴力,分类排序后枚举纵向的两条线段,扫描所有横向线段,记录和两条纵向都相交的num,计算ans即可
- 考虑对第三维的优化
- 确定x1较小的纵向线段之后,取出与它相交的横向线段,按x2排序。在枚举第二条纵向线段的时候依次删除这些横线段。f[y]表示y1=y的有效横线段数量,所求的就是两条纵向线段[y1,y2]交集的f[y]和
- 用线段树维护f数组,综合复杂度O(n^2 logn)
Code
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define ll long long
#define ls rt<<1
#define rs rt<<1|1
using namespace std;
int n,m,totz,toth,N;
ll hash[8007];
int num,last;
ll ans;
struct JOKER{
int x1,y1,x2,y2;
}z[2007],h[2007],t[2007];
struct joker{
int rt,l,r,s;
}tree[800007];
bool cmp(JOKER a,JOKER b){
//return a.x1<b.x1||a.x1==b.x1&&a.y1<b.y1||a.x1==b.x1&&a.y1==b.y1&&a.y2<.y2;
return a.x1<b.x1;
}
bool cmpwork(ll a,ll b){
return a<b;
}
bool cmpt(JOKER a,JOKER b){
return a.x2<b.x2;
}
int find(ll v){
int l=1,r=N,ans=0;
ll x=v;
while (l<=r){
int mid=(l+r)>>1;
if (x==hash[mid]){ans=mid; break;}
if (x>hash[mid]) l=mid+1; else r=mid-1;
}
return ans;
}
int erfen(int l,int r,int x){
if(l>r) return l;
int mid=(l+r)>>1;
if(x>t[mid].x2) return erfen(mid+1,r,x);
else return erfen(l,mid-1,x);
}
void work(){
int tot=0;
ll a[8007];
for (int i=1;i<=totz;i++){a[++tot]=z[i].y1; a[++tot]=z[i].y2;}
for (int i=1;i<=toth;i++){a[++tot]=h[i].y1; a[++tot]=h[i].y2;}
sort(a+1,a+tot+1,cmpwork);
N=1; hash[1]=a[1];
for (int i=2;i<=tot;i++)
if (a[i]!=a[i-1]) hash[++N]=a[i];
for (int i=1;i<=totz;i++){
z[i].y1=find(z[i].y1);
z[i].y2=find(z[i].y2);
}
for (int i=1;i<=toth;i++){
h[i].y1=find(h[i].y1);
h[i].y2=find(h[i].y2);
}
}
void build(int rt,int l,int r){
tree[rt].rt=rt;
tree[rt].r=r; tree[rt].l=l;
if (l==r){
tree[rt].s=0;
return;
}
int mid=(l+r)>>1;
build(ls,l,mid);
build(rs,mid+1,r);
tree[rt].s=tree[ls].s+tree[rs].s;
}
void insert(int rt,int id,int v){
int L=tree[rt].l,R=tree[rt].r;
if (L==R) {tree[rt].s+=v; return;}
int mid=(L+R)>>1;
if (id<=mid) insert(ls,id,v);
else insert(rs,id,v);
tree[rt].s=tree[ls].s+tree[rs].s;
}
ll query(int rt,int l,int r){
int L=tree[rt].l,R=tree[rt].r;
if (l<=L&&R<=r) return tree[rt].s;
int mid=(L+R)>>1; ll res=0;
if (l<=mid) res+=query(ls,l,r);
if (mid<r) res+=query(rs,l,r);
return res;
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++){
int x1,x2,y1,y2;
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
if (x1==x2){
z[++totz].x1=x1;
z[totz].x2=x2;
z[totz].y1=min(y1,y2);
z[totz].y2=max(y1,y2);
}
else{
h[++toth].x1=min(x1,x2);
h[toth].x2=max(x1,x2);
h[toth].y1=y1;
h[toth].y2=y2;
}
}
work();
sort(z+1,z+totz+1,cmp);
//sort(h+1,h+toth+1,cmp);
for (int i=1;i<totz;i++){
build(1,1,N);
num=0;
for (int ii=1;ii<=toth;ii++)
if (h[ii].x1<=z[i].x1&&h[ii].x2>=z[i].x1&&h[ii].y1>=z[i].y1&&h[ii].y1<=z[i].y2){
t[++num]=h[ii];
insert(1,h[ii].y1,1); // root 修改点 delta
}
sort(t+1,t+num+1,cmpt);
last=1;
for (int j=i+1;j<=totz;j++){
int k=erfen(last,num,z[j].x1);
for (int jj=last;jj<k;jj++) insert(1,t[jj].y1,-1);
last=k;
int up=max(z[i].y1,z[j].y1);
int down=min(z[i].y2,z[j].y2);
//if (up>down) swap(up,down);
ll s=0;
if (up<=down) s=query(1,up,down);
ans+=s*(s-1)/2;
//printf("ans=%d\n",ans);
}
}
printf("%lld",ans);
return 0;
}