机试练习06：poj3349——哈希表应用-程序员宅基地

一、题解方法

此题不能采用暴力枚举的办法，利用哈希表先将输入数据分类，然后针对key值相同的雪花，再进一步进行细致地判断。

key值采取雪花6个arm值加和，进行粗略比较，因为只有在key值相同的前提下，才有可能两个雪花完全一致。如果两个雪花key值相同，再分别比较每一个arm的长度，顺时针、逆时针比较判断二者异同。

只要发现存在相同的雪花，就返回结果。

二、题解代码

#include "stdio.h"
#include "stdlib.h"
#include <iostream>
#include "string.h"
#include "math.h"

using namespace std;

const int PRIME = 14997;


typedef struct
{
    int arm[6];
    
}leaves;

int hash_table[PRIME+3];
leaves snow[PRIME+3][100];

int getkey(leaves l)
{
    int i;
    int key = 0;
    for (i = 0; i < 6; i++)
    {
        key += l.arm[i];
    }
    key %= PRIME;
    return key;
}

int cmp(leaves x, leaves y)
{
    int st, i, j;
    for (st = 0; st < 6; st++){
        for (i = st, j = 0; j < 6; j++, i = (i + 1) % 6){
            if(x.arm[i] != y.arm[j]) break;
        }
        if(j == 6) return 1;
    }
    for (st = 0; st < 6; st++)
    {
        for (i = st, j = 0; j < 6; j++, i = (i + 5) % 6){
            if(x.arm[i] != y.arm[j]) break;
        }
        if(j == 6) return 1;
    }
    return 0;
}

int main()
{
    int n;
    
    scanf("%d", &n);
    memset(hash_table, 0, sizeof(hash_table));
    while(n--)
    {
        int i, pos;
        leaves leaf;
        for(i = 0; i < 6; i++)
        {
            scanf("%d", &leaf.arm[i]);
        }
        pos = getkey(leaf);
        for(i = 0; i < hash_table[pos]; i++)
        {
            if(cmp(leaf, snow[pos][i])){
                printf("Twin snowflakes found.");
                return 0;
            }
        }
        snow[pos][hash_table[pos]] = leaf;
        hash_table[pos]++;
    }
    printf("No two snowflakes are alike.");
    return 0;
}