POJ 2002 Squares

给出n个点，求出用这n个点可构成的正方形的个数。可以枚举两个点，求出正方形的另两个点。

然后判断这两个是否存在。我的hash公式写得比较烂，跑了1s多。

下面是求正方形剩下两点的公式：

已知： (x1,y1)  (x2,y2)

则：   x3=x1+(y1-y2)   y3= y1-(x1-x2)

x4=x2+(y1-y2)   y4= y2-(x1-x2)

或

x3=x1-(y1-y2)   y3= y1+(x1-x2)

x4=x2-(y1-y2)   y4= y2+(x1-x2)

/*Accepted    1724K    1047MS    C++    1713B    2012-08-24 11:46:05*/

#include<stdio.h>

#include<string.h>

#include<stdlib.h>



const int MAXN = 1 << 10;

const int prime = 99991;

int n, ans;



struct point

{

    int x, y;

}t[MAXN];



struct Hash

{

    point p;

    struct Hash *next;

}h[prime + 10];



int hash(point a)

{

    return (a.x * a.x + a.y * a.y) % prime + 1;

}



void insert(point a)

{

    Hash *temp = new Hash();

    int key = hash(a);

    temp->p.x = a.x;

    temp->p.y = a.y;

    temp->next = h[key].next;

    h[key].next = temp;

}



bool exist(point a)

{

    Hash *temp = h[hash(a)].next;



    while(temp)

    {

        if(temp->p.x == a.x && temp->p.y == a.y)

            return true;

        temp = temp->next;

    }

    return false;

}



void Read()

{

    int i;

    for(i = 0; i < n; i ++)

    {

        scanf("%d%d", &t[i].x, &t[i].y);

        insert(t[i]);

    }

}



void cal()

{

    point a, b;

    int i, j;

    ans = 0;

    for(i = 0; i < n - 1; i ++)

        for(j = i + 1; j < n; j ++)

        {

            a.x = t[i].y - t[j].y + t[i].x;

            a.y = t[i].y - (t[i].x - t[j].x);

            b.x = t[j].x + t[i].y - t[j].y;

            b.y = t[j].y - (t[i].x - t[j].x);

            if(exist(a) && exist(b))

                ans ++;



            a.x = t[i].x - (t[i].y - t[j].y);

            a.y = t[i].y + t[i].x - t[j].x;

            b.x = t[j].x - (t[i].y - t[j].y);

            b.y = t[j].y + t[i].x - t[j].x;

            if(exist(a) && exist(b))

                ans ++;

        }

    ans /= 4;

}



int main()

{

    while(scanf("%d", &n), n)

    {

        memset(h, 0, sizeof h);

        Read();

        cal();

        printf("%d\n", ans);



    }

    return 0;

}