codeforces 425D Sereja and Squares n个点构成多少个正方形

输入n个点，问可以构成多少个正方形。n,xi,yi<=100,000。

 

刚看题的时候感觉好像以前见过╮(╯▽╰)╭最近越来越觉得以前见过的题偶尔就出现类似的，可是以前不努力啊，没做出来的没认真研究

首先想到的朴素是n^2的算法，10^10显然不行=。=抱着过预判的侥幸心理写了一发，打算过预判之后锁上看别人代码怎么做，结果TLE 13

第二天早上起来看题解，看不懂，看别人的代码才懂的。。。。

大概就是，避开最坏情况，枚举正方形左下角的点，然后如果当前x的点数量比sqrt（n）要小，就用当前x的上边的点做正方形的左上角，再检查另外2个点是否存在就好了；如果当前x的点数量比sqrt（n）要大，就用当前y的右边的点做正方形的右下角，再检查。

复杂度大概就是nlogn吧

拖了这么久终于写下第一篇博客日志了~~~

 

哪里有错的话求指正。。。。。

 

Note:正方形要平行于坐标轴

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <string>
#include <vector>
#include <queue>
#include <set>
using namespace std;

#define ll long long
#define inf 0x3f3f3f3f
#define eps 1e-8

const int maxn = 100010;
const int sqrtmaxn = (int)sqrt(maxn+.0) + 1;
vector<int>vec[maxn];
//set<int>st[maxn*2];
inline bool find(int x,int y){
    if(x>=maxn)return false;
    return binary_search(vec[x].begin(),vec[x].end(),y);
}
int main(){
    int n;
    while(~scanf("%d",&n)){
        //for(int i=0;i<maxn;++i)vec[i].clear(),st[i].clear();
        for(int i=0;i<n;++i){
            int x,y;
            scanf("%d%d",&x,&y);
            vec[x].push_back(y);
            //st[x].insert(y);
        }
        for(int i=0;i<maxn;++i)sort(vec[i].begin(),vec[i].end());
        ll ans=0;
        for(int x=0;x<maxn;++x){
            if(vec[x].size()<sqrtmaxn){
                for(int i=0;i<vec[x].size();++i){
                    for(int j=i+1;j<vec[x].size();++j){
                        int d=vec[x][j]-vec[x][i];
                        if(find(x+d,vec[x][i]) && find(x+d,vec[x][j]))++ans;
                    }
                }
            }
            else {
                for(int xx=x+1;xx<maxn;++xx){
                    for(int i=0;i<vec[xx].size();++i){
                        int yy=vec[xx][i],d=xx-x;
                        if(find(x,yy) && find(x,yy+d) && find(xx,yy+d))++ans;
                    }
                }
            }
        }
        printf("%I64d\n",ans);
    }
    return 0;
}