poj - 3349 Snowflake Snow Snowflakes（Hash）

题目链接：http://poj.org/problem?id=3349

题意：给你n片雪花，每一片雪花有6个花瓣，每个花瓣有一个值，两片雪花完全相同，当且仅当这两片雪花的六片花瓣循环左移或者循环右移后完全相同，最后问你有没有两片雪花是完全相同的。

思路：hash，首先建立hash函数key = (a[0]+a[1]+a[2]+a[3]+a[4]+a[5])%prime。由于题目给的n的范围是1e5，所以我们可以去1e5范围内最大的素数99991作为那个prime这样，如果两片雪花相同，那么这两片雪花的hash值就一定相同，这样，我们可以用拉链法将hash相同的雪花放在一起，每一次暴力的遍历所有hash值一样的，如果找到一样的雪花，则输出：Twin snowflakes found.，反之，则输出：No two snowflakes are alike.

AC代码：

#include <map>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define LL long long

const int maxn = 1e5 + 7;
const int mod = 99991;
const LL INF = 1e18;

int a[maxn][6];
vector<int>v[maxn];

bool Judge(int x1,int x2){
    for(int i=0;i<6;i++){
        if((
            a[x1][0]==a[x2][i]&& a[x1][1]==a[x2][(i+1)%6]&&
            a[x1][2]==a[x2][(i+2)%6]&& a[x1][3]==a[x2][(i+3)%6]&&
            a[x1][4]==a[x2][(i+4)%6]&& a[x1][5]==a[x2][(i+5)%6])
        ||(   
           a[x1][0]==a[x2][i]&& a[x1][1]==a[x2][(i+5)%6]&&
           a[x1][2]==a[x2][(i+4)%6]&& a[x1][3]==a[x2][(i+3)%6]&&
           a[x1][4]==a[x2][(i+2)%6]&& a[x1][5]==a[x2][(i+1)%6]))
        return true;
    }
    return false;
}

int main() {
    int n;
    while(~scanf("%d",&n)) {
        for(int i = 0 ; i < n ; i++) {
            for(int j = 0 ; j < 6 ; j++) {
                scanf("%d",&a[i][j]);
            }
        }
        for(int i = 0 ; i < maxn ; i++) v[i].clear();
        bool flag = false;
        for(int i = 0 ; i < n ; i++) {
            int sum = 0;
            for(int j = 0 ; j < 6 ; j++) {
                sum = (sum%mod + a[i][j]%mod)%mod;
            }
            for(int j = 0 ; j < v[sum].size() ; j++) {
                if(Judge(v[sum][j],i)) {
                    flag = true;
                    break;
                }
            }
            v[sum].push_back(i);
        }
        if(flag) printf("Twin snowflakes found.\n");
        else printf("No two snowflakes are alike.\n");
    }
    return 0;
}