C 雪花算法

1. #include<stdio.h>  
2. #include<stdlib.h>;  
3. #include<algorithm>  
4. #define inf 100001  
5. using namespace std;   
6. struct Edge{  
7.     int a[10],next;  
8. }edge[100100];  
9. int a[10],head[100100],cnt;  
10. bool add(int * b){  
11.     int sum=0,i,j,p,q;  
12.     for(i=0;i<6;i++)  
13.         sum+=b[i];  
14.     int hash=sum%inf;  
15.     for(i=head[hash];i;i=edge[i].next){  
16.         for(j=0;j<6;j++){  
17.             if(edge[i].a[j]==b[0]){  
18.                 for(p=0,q=j;p<6;p++){  
19.                     if(b[p]!=edge[i].a[q])  
20.                         break;  
21.                         q=(q+1)%6;  
22.                 }  
23.                 if(p==6)return 1;  
24.                 for(p=0,q=j;p<6;p++){  
25.                     if(b[p]!=edge[i].a[q])  
26.                         break;  
27.                     q=(q-1+6)%6;  
28.                 }  
29.                 if(p==6)return 1;  
30.             }  
31.         }  
32.     }  
33.     ++cnt;  
34.     for(i=0;i<6;i++)  
35.         edge[cnt].a[i]=b[i];  
36.     edge[cnt].next=head[hash];  
37.     head[hash]=cnt;  
38.     return 0;  
39. }  
40. int main(){  
41.     int i,j,n;  
42.     scanf("%d",&n);  
43.     bool flag=0;  
44.     for(i=1;i<=n;i++){  
45.         for(j=0;j<6;j++){  
46.             scanf("%d",&a[j]);  
47.         }  
48.         if(flag)continue;  
49.         if(add(a))flag=1;  
50.     }  
51.     if(flag)              
52.         printf("Twin snowflakes found.\n");          
53.     else              
54.         printf("No two snowflakes are alike.\n");  
55. }