BOJ)2171 직사각형의 개수

문제: icpc.me/2171

2차원 평면위에 5000개 이하의 점이 있을 때 점들로 만들 수 있는 모든 직사각형의 수를 세는 문제이다,

n제한이 5000에 2초이므로 n^2logn 솔루션으로 풀 수 있을 것이라고 생각하였다.

n*n으로 모든 두 점에 대하여 직사각형의 대각선에 놓이는 경우를 생각하면, 나머지 대각선에 포함되는 두 점은 set을 이용해 존재여부를 log시간에 찾을 수 있다.

#include <cstdio>
#include <algorithm>
#include <set>
#include <vector>
using namespace std;
int n,res;
set<pair<int,int>> st;
vector<pair<int,int>> vt;
int main(){
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        vt.push_back({x,y});
        st.insert({x,y});
    }
    for(int i=0;i<n;i++){
        for(int j=i+1;j<n;j++){
            if(vt[i].first!=vt[j].first&&vt[i].second!=vt[j].second){
                if(st.count({vt[i].first,vt[j].second})&&st.count({vt[j].first,vt[i].second}))
                    res++;
            }
        }
    }
    printf("%d\n",res/2);
    return 0;
}