如果可能的话, kd 树肯定是要走的路,但问题是您正在寻找“给定半径内的所有帖子”,其中可能有很多(100 到甚至 10000)。但是,您可能还需要考虑集群化帖子以避免高度集中的情况,以及其他一些好处,代价是半径(5 公里)是近似值。可以做到这一点的一种方法是使用最小圆的线性算法。
def cluster_posts(points,cluster_radius):
clusters = dict()
for p in posts:
# This inner part is also done whenever a new post is added
clusters[p] = Cluster([p])
points_set = set(points)
While points_set:
# This inner part is also done whenever a new post is added
p = points_set.pop()
q = kd_tree.nearest_neighbor(p)
dist = distance(p,q)
radius_p = clusters[p].smallest_circle_radius()
radius_q = clusters[q].smallest_circle_radius()
if radius_p + dist + radius_q < cluster_radius:
new_cluster = clusters[p].merge(clusters[q])
r = new_cluster.smallest_circle_radius()
if r < cluster_radius:
c = new_cluster.smallest_circle_center()
points_set.remove(q)
clusters.remove(q)
clusters.remove(p)
kd_tree.remove(p)
kd_tree.remove(q)
points_set.add(c)
kd_tree.add(c)
clusters[c] = new_cluster
以上尝试根据 cluster_radius 将两个集群组合成一个集群。有一些优化空间,但它应该在 O(N log N) 左右运行。由于我没有编写某些类和函数的代码,因此它不会编译,但希望它能够传达这个想法。它假设点(帖子的纬度/经度)已经输入到 kd 树中。将 lat 和 long 从 degree-minutes-seconds.fractions 转换为 seconds.fractions 也可能不是一个坏主意。5 km 可能足够小,可以将坐标视为欧几里得点而不会引入太多误差,因为无论如何它都是聚类的近似值。
查询只是通过kd树找到用户位置(query_radius - cluster_radius)内的所有簇,并且至少包括最近的簇。您提供的数字将使查询半径为 5 公里。集群半径的几种可能性:
- 5公里(全半径)
- 2.5公里(半半径)
- 几百英尺(按块排列)