首页 > 代码库 > POJ 1021 人品题

POJ 1021 人品题

报告见代码。。

  1 #include <iostream>
  2 #include <cstdio>
  3 #include <cstring>
  4 #include <algorithm>
  5 using namespace std;
  6 
  7 const int MAX=105;
  8 int dir[4][2]={1,0,-1,0,0,1,0,-1};
  9 int h,w,n;
 10 int maze[MAX][MAX];
 11 int maze2[MAX][MAX];
 12 struct{
 13     int x,y;
 14 }save[1000];
 15 int sa;
 16 int h1[1000];
 17 int h2[1000];
 18 
 19 void dfs(int i,int j,int (*p)[MAX]){
 20     p[i][j]=0;
 21     save[++sa].x=i;save[sa].y=j;
 22     for(int k=0;k<4;k++){
 23         int tx,ty;
 24         tx=i+dir[k][0];
 25         ty=j+dir[k][1];
 26         if(p[tx][ty]==1&&tx<h&&tx>=0&&ty<w&&ty>=0)
 27         dfs(tx,ty,p);
 28     }
 29 }
 30 
 31 void calculate(int *ha,int &top){   //HASH函数是各点对的距离的平方。。。也过。 
 32     int i,j,x=0,y=0; top++;
 33     for(i=1;i<=sa;i++){
 34         x=save[i].x;
 35         y=save[i].y;
 36         for(j=i+1;j<=sa;j++){
 37             int dx=abs(x-save[j].x);
 38             int dy=abs(y-save[j].y);
 39             int dis=dx*dx+dy*dy;
 40             ha[top]+=dis;
 41         }
 42     }
 43 }
 44 
 45 void slove(int (*p)[MAX],int *ha,int &top){
 46     int i,j;
 47     for(i=0;i<h;i++){
 48         for(j=0;j<w;j++){
 49             sa=0;
 50             if(p[i][j]==1){
 51             dfs(i,j,p);
 52             calculate(ha,top);
 53             }
 54         }
 55     }
 56 }
 57 
 58 int main(){
 59     int x,y,cas;
 60     scanf("%d",&cas);
 61     while(cas--){
 62         scanf("%d%d%d",&w,&h,&n);
 63         memset(maze,0,sizeof(maze));
 64         memset(maze2,0,sizeof(maze2));
 65         memset(h1,0,sizeof(h1));
 66         memset(h2,0,sizeof(h2));
 67         int top1=-1,top2=-1;
 68         for(int i=1;i<=n;i++){
 69             scanf("%d%d",&x,&y);
 70             maze[y][x]=1;
 71         }
 72     /*    for(int i=0;i<h;i++){
 73             for(int j=0;j<w;j++)
 74             printf("%d ",maze[i][j]);
 75             printf("\n");
 76         }*/
 77         for(int i=1;i<=n;i++){
 78             scanf("%d%d",&x,&y);
 79             maze2[y][x]=1;
 80         }
 81      /*   for(int i=0;i<h;i++){
 82             for(int j=0;j<w;j++)
 83             printf("%d ",maze2[i][j]);
 84             printf("\n");
 85         }*/
 86         slove(maze,h1,top1);
 87         slove(maze2,h2,top2);
 88         if(top1!=top2){
 89             printf("NO\n");
 90             continue;
 91         }
 92         bool flag=true;
 93         sort(h1,h1+top1+1);
 94         sort(h2,h2+top2+1);
 95         for(int i=0;i<=top1;i++){
 96         //    printf("%d %d \n",h1[i],h2[i]);
 97             if(h1[i]!=h2[i]){
 98                 flag=false;
 99                 break;
100             }
101         }
102         if(!flag)
103         printf("NO\n");
104         else
105         printf("YES\n");
106     }
107     return 0;
108 }
View Code