Codeforces 1194E Count The Rectangles

数据结构 专栏收录该内容
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;
}

 

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值