POJ 2002 Squares(hash+二分查找)

Description
给出n个整点的坐标,问这n个点能组成多少个正方形
Input
多组用例,每组用例第一行为点数n,之后n行每行两个整数x和y表示该点坐标,以n=0结束输入(1 <= n <= 1000,|x|,|y|<20000)
Output
对于每组用例,输出这n个点可以组成多少个正方形
Sample Input
4
1 0
0 1
1 1
0 0
9
0 0
1 0
2 0
0 2
1 2
2 2
0 1
1 1
2 1
4
-2 5
3 7
0 0
5 2
0
Sample Output
1
6
1
Solution
显然每给出两点坐标,都能求出两组另外两点坐标,而枚举两点去暴力查找显然不可行,所以首先可以用hash将一个点的坐标表示成一个整数,然后就可以二分查找一个点是否在这n个点中存在,注意这样求出的数量是真实数量的四倍,所以除以四才是答案
Code

#include<stdio.h>
#include<string.h>
#define maxn 1111
#define mod 10007
struct node
{
    int x,y,next;
}edge[maxn];
int x[maxn],y[maxn],tot,hash[mod],n;
long long ans;
void init()
{
    memset(hash,-1,sizeof(hash));
    tot=0;
    ans=0;
}
void Go_Hash(int x,int y)
{
    edge[tot].x=x;
    edge[tot].y=y;
    int h=(x*x+y*y)%mod;
    edge[tot].next=hash[h];
    hash[h]=tot++;
}
int Go_Search(int x,int y)
{
    int h=(x*x+y*y)%mod;
    for(int i=hash[h];~i;i=edge[i].next)
        if(edge[i].x==x&&edge[i].y==y)return 1;
    return 0;
}
int main()
{
    while(~scanf("%d",&n),n)
    {
        init();
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&x[i],&y[i]);
            Go_Hash(x[i],y[i]);
        }
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++)
            {
                int x1=x[i],y1=y[i],x2=x[j],y2=y[j],x3,y3,x4,y4;
                x3=x1+(y1-y2);y3=y1-(x1-x2);
                x4=x2+(y1-y2);y4=y2-(x1-x2);
                if(Go_Search(x3,y3)&&Go_Search(x4,y4))ans++;
            }
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++)
            {
                int x1=x[i],y1=y[i],x2=x[j],y2=y[j],x3,y3,x4,y4;
                x3=x1-(y1-y2);y3=y1+(x1-x2);
                x4=x2-(y1-y2);y4=y2+(x1-x2);
                if(Go_Search(x3,y3)&&Go_Search(x4,y4))ans++;
            }
        printf("%lld\n",ans>>2);
    } 
    return 0;
} 

你可能感兴趣的:(POJ 2002 Squares(hash+二分查找))