POJ 1021 人品题

报告见代码。。

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

 const int MAX=;
 int dir[][]={,,-,,,,,-};
 int h,w,n;
 int maze[MAX][MAX];
 int maze2[MAX][MAX];
 struct{
     int x,y;
 }save[];
 int sa;
 int h1[];
 int h2[];

 void dfs(int i,int j,int (*p)[MAX]){
     p[i][j]=;
     save[++sa].x=i;save[sa].y=j;
     for(int k=;k<;k++){
         int tx,ty;
         tx=i+dir[k][];
         ty=j+dir[k][];
         if(p[tx][ty]==&&tx<h&&tx>=&&ty<w&&ty>=)
         dfs(tx,ty,p);
     }
 }

 void calculate(int *ha,int &top){   //HASH函数是各点对的距离的平方。。。也过。
     int i,j,x=,y=; top++;
     for(i=;i<=sa;i++){
         x=save[i].x;
         y=save[i].y;
         for(j=i+;j<=sa;j++){
             int dx=abs(x-save[j].x);
             int dy=abs(y-save[j].y);
             int dis=dx*dx+dy*dy;
             ha[top]+=dis;
         }
     }
 }

 void slove(int (*p)[MAX],int *ha,int &top){
     int i,j;
     for(i=;i<h;i++){
         for(j=;j<w;j++){
             sa=;
             if(p[i][j]==){
             dfs(i,j,p);
             calculate(ha,top);
             }
         }
     }
 }

 int main(){
     int x,y,cas;
     scanf("%d",&cas);
     while(cas--){
         scanf("%d%d%d",&w,&h,&n);
         memset(maze,,sizeof(maze));
         memset(maze2,,sizeof(maze2));
         memset(h1,,sizeof(h1));
         memset(h2,,sizeof(h2));
         int top1=-,top2=-;
         for(int i=;i<=n;i++){
             scanf("%d%d",&x,&y);
             maze[y][x]=;
         }
     /*    for(int i=0;i<h;i++){
             for(int j=0;j<w;j++)
             printf("%d ",maze[i][j]);
             printf("\n");
         }*/
         for(int i=;i<=n;i++){
             scanf("%d%d",&x,&y);
             maze2[y][x]=;
         }
      /*   for(int i=0;i<h;i++){
             for(int j=0;j<w;j++)
             printf("%d ",maze2[i][j]);
             printf("\n");
         }*/
         slove(maze,h1,top1);
         slove(maze2,h2,top2);
         if(top1!=top2){
             printf("NO\n");
             continue;
         }
         bool flag=true;
         sort(h1,h1+top1+);
         sort(h2,h2+top2+);
         for(int i=;i<=top1;i++){
         //    printf("%d %d \n",h1[i],h2[i]);
             if(h1[i]!=h2[i]){
                 flag=false;
                 break;
             }
         }
         if(!flag)
         printf("NO\n");
         else
         printf("YES\n");
     }
     return ;
 }