题目大意:给出几个雪花的六个手臂长度,可能以顺时针或逆时针的顺序给出,但是不一定从何处开始,求是否有相同的雪花
输入:雪花个数n( 0 < n ≤ 100000)
第i个雪花的六个手臂长度(手臂长度从0到10000000)
输出:如果有两个相同的雪花则输出Twin snowflakes found.
如果所有雪花都不同则输出No two snowflakes are alike.
分析:如果两两比较复杂度为n²超时,所以用哈希,找到同类后再两两比较。哈希函数为六个手臂长度和再模质数,因为最多有100000个雪花,所以取模质数为99991(事实上在100000周围的质数都可以),若查找到相同key值的雪花,再进行具体手臂长度的比较。
这里面比较两个雪花是否完全相同,除了代码中给出的方法之外,还有一种方法:
sort(arm[a],arm[a]+6);
sort(arm[b],arm[b]+6);
for(int i = 0;i < 6;i++)
if(arm[a][i] != arm[b][i])
return false;
return true;
也就是将雪花的六个手臂从小到大排序之后,每个手臂都相等才满足两个雪花完全相同
代码:转载自https://www.cnblogs.com/fu11211129/p/4211506.html
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<map>
#include<vector>
using
namespace
std;
const
int
max_size=100010;
const
int
prime=99991;
vector<
int
> hsh[prime];
int
arm[max_size][6];
int
n;
bool
isTheSame(
int
a,
int
b){
for
(
int
i=0;i<6;i++){
if
((
//clockwise
arm[a][0]==arm[b][i]&& arm[a][1]==arm[b][(i+1)%6]&&
arm[a][2]==arm[b][(i+2)%6]&& arm[a][3]==arm[b][(i+3)%6]&&
arm[a][4]==arm[b][(i+4)%6]&& arm[a][5]==arm[b][(i+5)%6])
||
//counter clockwise
( arm[a][0]==arm[b][i]&& arm[a][1]==arm[b][(i+5)%6]&&
arm[a][2]==arm[b][(i+4)%6]&& arm[a][3]==arm[b][(i+3)%6]&&
arm[a][4]==arm[b][(i+2)%6]&& arm[a][5]==arm[b][(i+1)%6]))
return
true
;
}
return
false
;
}
int
main(){
scanf
(
"%d"
,&n);
long
long
sum,key;
for
(
int
i=0;i<n;i++){
for
(
int
j=0;j<6;j++){
scanf
(
"%d"
,&arm[i][j]);
}
}
for
(
int
i=0;i<n;i++){
sum=0;
for
(
int
j=0;j<6;j++){
sum+=arm[i][j];
}
key=sum%prime;
int
k;
for
(k=0;k<hsh[key].size();k++){
if
(isTheSame(i,hsh[key][k])){
printf
(
"Twin snowflakes found.\n"
);
exit
(0);
}
}
hsh[key].push_back(i);
}
printf
(
"No two snowflakes are alike.\n"
);
}