正方形个数(二维点哈希)

题目：http://poj.org/problem?id=2002

 

题意：平面上给定n个点，求由这些点能构成多少个正方形。

 

分析：我们枚举两个点，那么在这两个点形成的直线两边可以计算出正方形的另外两个点，利用Hash然后查找即可。

 

#include <iostream>
#include <string.h>
#include <stdio.h>

using namespace std;
typedef long long LL;
const int N = 1005;
const int H = 10007;

struct Point
{
    int x,y;
};

Point p[N];

struct Node
{
    int x,y;
    int next;
};

Node node[N];
int cur,n;
LL ans;

int Hash[H];

void Init()
{
    for(int i=0;i<H;i++)
       Hash[i] = -1;
    cur = ans = 0;
}

void Insert(int x,int y)
{
    int h = (x * x + y * y) % H;
    node[cur].x = x;
    node[cur].y = y;
    node[cur].next = Hash[h];
    Hash[h] = cur++;
}

bool Search(int x,int y)
{
    int h = (x * x + y * y) % H;
    int t = Hash[h];
    while(t != -1)
    {
        if(x == node[t].x && y == node[t].y) return true;
        t = node[t].next;
    }
    return false;
}

LL Work()
{
    Init();
    for(int i=0;i<n;i++)
    {
        scanf("%d%d",&p[i].x,&p[i].y);
        Insert(p[i].x,p[i].y);
    }
    for(int i=0;i<n;i++)
    {
        for(int j=i+1;j<n;j++)
        {
            int x1 = p[i].x - (p[i].y - p[j].y);
            int y1 = p[i].y + (p[i].x - p[j].x);
            int x2 = p[j].x - (p[i].y - p[j].y);
            int y2 = p[j].y + (p[i].x - p[j].x);
            if(Search(x1,y1) && Search(x2,y2)) ans++;
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=i+1;j<n;j++)
        {
            int x1 = p[i].x + (p[i].y - p[j].y);
            int y1 = p[i].y - (p[i].x - p[j].x);
            int x2 = p[j].x + (p[i].y - p[j].y);
            int y2 = p[j].y - (p[i].x - p[j].x);
            if(Search(x1,y1) && Search(x2,y2)) ans++;
        }
    }
    ans >>= 2;
    return ans;
}

int main()
{
    while(~scanf("%d",&n),n)
        printf("%I64d\n",Work());
    return 0;
}

 

题目：http://acm.hust.edu.cn/problem.php?id=1623

 

分析：本题与上题一样，只是注意本题中的点可能重合，所以一定要先去重。