Codeforces 1194E Count The Rectangles

最新推荐文章于 2019-07-24 09:27:17 发布

Gipsyu

最新推荐文章于 2019-07-24 09:27:17 发布

阅读量282

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/Gipsy_Danger/article/details/95938274

版权

数据结构专栏收录该内容

54 篇文章 0 订阅

订阅专栏

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M=5e3+1,N=1e4+1;
vector<array<int,3> >V,H;
int T[N+10];
int lowbit(int x){return x&(-x);}
void add(int x,int val){for(;x<=N;x+=lowbit(x))T[x]+=val;}
int ask(int x){
    int ans=0;
    for(;x;x-=lowbit(x))ans+=T[x];
    return ans;
}
int main(){
    int n,x1,y1,x2,y2;cin>>n;ll ans=0;
    for(int i=1;i<=n;++i){
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        x1+=M;x2+=M;y1+=M;y2+=M;
        if(x1>x2)swap(x1,x2);
        if(y1>y2)swap(y1,y2);
        if(x1==x2)V.push_back({x1,y1,y2});
        if(y1==y2)H.push_back({y1,x1,x2});
    }
    sort(H.begin(),H.end());
    sort(V.begin(),V.end(),[](array<int,3> a,array<int,3> b){return a[2]<b[2];});
    for(int i=0;i<H.size();++i){
        memset(T,0,sizeof T);
        for(int j=0;j<V.size();++j)if(V[j][1]<=H[i][0])add(V[j][0],1);
        for(int j=i+1,k=0;j<H.size();){
            if(k<V.size()&&V[k][2]<H[j][0]){
                if(V[k][1]<=H[i][0])add(V[k][0],-1);
                ++k;
            }else{
                int l=max(H[i][1],H[j][1]);int r=min(H[i][2],H[j][2]);
                if(l<r){
                    int sum=ask(r)-ask(l-1);
                    ans=ans+sum*(sum-1)/2;
                }
                ++j;
            }
        }
    }
    cout<<ans<<endl;
}