题目
- 给定平面上 n 对不同的点,“回旋镖” 是由点表示的元组 (i, j, k) ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。
- 找到所有回旋镖的数量。你可以假设 n 最大为 500,所有点的坐标在闭区间 [-10000, 10000] 中。
示例
①示例1
- 输入: [[0,0],[1,0],[2,0]]
- 输出: 2
- 解释: 两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]
说明
①相关话题
②题目地址
解题方法
①哈希表 + 数学
- 这里的哈希表直接使用 uthash,节省空间。
- uthash 是一个用 C 语言编写的开源库,使用宏实现了哈希表的增删改查等功能。
- 题中说明需要考虑元组顺序,意味着若有 n 个点与 A 点的距离值都等于 m,则这 n 个点与 A 点有 n(n-1) 种符合题意的排列组合方式,即有 n(n-1) 个回旋镖。
- 所以对于每个点,我们计算它到其他点的距离,并将距离作为键值保存在哈希表中,同时要记录不同键值(距离)的出现次数(n),然后遍历哈希表计算每个键值(距离)的 n(n-1) 并求和,可得到每个点的总回旋镖数,最后再对所有点求和即可。
- 时间复杂度:O(N^2)。
- 空间复杂度:O(N)。
代码详解
int numberOfBoomerangs(int** points, int pointsRowSize, int pointsColSize) {
struct hash {
int key;
int count;
UT_hash_handle hh;
};
int count = 0;
for (int i = 0; i < pointsRowSize; i++) {
struct hash *hashTable = NULL;
for (int j = 0; j < pointsRowSize; j++) {
if (j != i) {
struct hash *h;
int p = points[i][0]-points[j][0], q = points[i][1]-points[j][1];
int n = p*p + q*q;
HASH_FIND_INT(hashTable, &n, h);
if (h)
h->count++;
else {
h = malloc(sizeof(struct hash));
h->key = n;
h->count = 1;
HASH_ADD_INT(hashTable, key, h);
}
}
}
for (struct hash *s = hashTable; s!= NULL; s = s->hh.next)
count += s->count*(s->count-1);
}
return count;
}
附录