pku 2002 Squares

http://poj.org/problem?id=2002 

用二维hash数组对输入的点进行标记。对原数组进行排序,然后从前到后任选俩个点,作为一个正方形的边,然后再根据关系求出另外两个点,在hash数组里看是否进行了标记,如果有,这找到了一个正方形。

 1 #include<stdio.h>
2 #include<string.h>
3 #include<iostream>
4 #include<stdlib.h>
5 #include<algorithm>
6 using namespace std;
7 #define N 2010
8 #define K 1000
9 int n;
10 struct node
11 {
12 int x,y;
13 }a[N];
14 int hash[2*N][2*N];
15 int cmp(node a,node b)
16 {
17 if(a.x!=b.x) return a.x<b.x;
18 return a.y<b.y;
19 }
20 int chang(node g,node b)
21 {
22 int i,j,t;
23 node c,d;
24 t=(g.x-b.x)*(g.y-b.y);
25 if(t>0||g.y==b.y) return 0;
26 i=abs(g.x-b.x);
27 j=abs(g.y-b.y);
28 c.x=g.x+j;
29 c.y=g.y+i;
30 d.x=b.x+j;
31 d.y=b.y+i;
32 if(hash[c.x+K][c.y+K]&&hash[d.x+K][d.y+K]) return 1;
33 else return 0;
34 }
35 int main()
36 {
37 int i,j;
38 while(cin>>n,n)
39 {
40 //memset(hash,0,sizeof(hash)); 如果用memset是1641MS。
41 for(i=1;i<=n;i++) //但是只要在下面修改一下就是在最后对用过的hash再恢复原先的值,则用时297MS。而且占用内存好像也少了
42 {
43 //cin>>a[i].x>>a[i].y;
44 scanf("%d%d",&a[i].x,&a[i].y);
45 hash[K+a[i].x][K+a[i].y]=1; //加上K 是因为坐标最小值是-1000,所以这里就是把所有的都转化为正数
46 }
47 sort(a+1,a+n+1,cmp);
48 int ans=0;
49 for(i=1;i<=n;i++)
50 {
51 for(j=i+1;j<=n;j++)
52 {
53 if(chang(a[i],a[j])) ans++;
54 }
55 }
56 cout<<ans<<endl;
57 for(i=1;i<=n;i++)
58 hash[a[i].x+K][a[i].y+K]=0;
59 }
60 return 0;
61 }


 

posted @ 2012-03-09 19:03  AC_Girl  阅读(137)  评论(0编辑  收藏  举报