POJ 2002 (hash + 几何)

算法过程：

1 将顶点按x坐标递增排序，若x相同，按y坐标递增排序，然后枚举所有边，对每一条由点p1和p2(根据排序p1 < p2)组成的边按照如下方式可唯一确定一个正方形：

    1) 将边绕p1逆时针旋转90度得到点p3

    2) 将边绕p2顺时针旋转90度得到点p4

则p1 p2 p3 p4组成一个正方形，设p1 = (x1,y1), p2 = (x2, y2)，根据向量的旋转公式可以求出p3, p4的坐标为

p3 = (y1 - y2 + x1, x2 - x1 + y1)

p4 = (y1 - y2 + x2, x2 - x1 + y2)

 

2 然后搜索点p3和p4是否存在，若存在则找到一个正方形，计数加1，可以发现总是存在两条边确定的正方形是一样的，也就是说每个正方形会被发现2次，所以要将最后的计数结果除以2.

#include <iostream>
#include <algorithm>
using namespace std;

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

const int POINT_NUM = 1002;
const int MAX_NUM = 20002;
const int INF = 0x3F3F3F3F;
point pt[POINT_NUM];
int head[MAX_NUM];
int n, base;

bool cmp(const point& a, const point& b)
{
	if (a.x < b.x) 
		return true;
	else if (a.x == b.x) 
		return a.y < b.y;
	else
		return false;
}

void make_hash()
{
	int i, code;
	sort(pt, pt+n, cmp); //将顶点按x坐标递增排序,若x相同,按y坐标递增排序 
	for (i = 0; i < n; ++i){
		code = pt[i].y - base;  //根据y做hash
		pt[i].next = head[code];
		head[code] = i;
	}
}

bool find(int x, int y)
{
	int i, code = y - base;
	for (i = head[code]; i != -1 && pt[i].x != x; i = pt[i].next);
	return i != -1;
}

int solve()
{
	int i, j, sum = 0;
	//枚举任意两个点组成正方形一条边，计算出另两个点，在hash表中找，找到则sum加1
	for (i = 0; i < n; ++i){
		for (j = i+1; j < n; ++j){
			int x1 = pt[i].y - pt[j].y + pt[i].x;
			int y1 = pt[j].x - pt[i].x + pt[i].y;
			int x2 = pt[i].y - pt[j].y + pt[j].x;
			int y2 = pt[j].x - pt[i].x + pt[j].y;
			if (find(x1, y1) && find(x2, y2)) sum++;
		}
	}
	return sum;
}

int main()
{
	while (scanf("%d", &n) && n != 0){
		base = INF;
		memset(head, -1, sizeof(head));
		for (int i = 0; i < n; ++i){
			scanf("%d%d", &pt[i].x, &pt[i].y);
			if (pt[i].y < base) base = pt[i].y;
		}
		make_hash();
		printf("%d\n", solve()/2);//每个正方形被算了两次
	}
	return 0;
}