problem2:
已知一个数d(0< d < 1 比如取d = 0.01),在单位正方形(就是指边长为1)中产生N(100000左右)个随机的点集合,求有多少点对可以用长度小于d的直线连接。
首先problem2来源于一个现实点的问题,比如在中国这张地图上,寻找相邻距离小于d的城市对。大概思路就是将每个城市用坐标表示,将最小坐标和最大坐标连成一个矩形.把各个城市看做矩形内的点,寻找可以用长度不超过d的线段连接的点对。为了方便测试,我把整个地图压缩成一个单位正方形,在里面随机的产生N的坐标点来模拟城市。于是就得到了问题problem2.
一种比较直观的解法就是遍历每个点对,判断他们之间的距离是否小于d。这样实现起来很方便,时间复杂度为O(N^2),对于较大的N性能太差。但事实上,很难找到一个降低时间复杂度的算法(至少我没有找到),但是在判断点对的时候却可以利用剪裁技术,使程序运行得很快。
具体的剪裁思路:
首先根据d,将整个地图划分为边长为d的正方形网格,注意,寻找一个点满足条件的点对时候,并不需要扫描所有的点,只需要判断该点所在网格和其相邻的网格中点是否满足条件即可,从而剪裁掉其他网格内的点。这种技术在d远小于地图的边长时效果非常的好,比如距离d只有中国南北距离的1/100.根据这个思路,我写了个程序来,取N=100000,d=0.01时,运行效果不错(1s以内)。而不用剪裁直接遍历所有点对的话基本上需要几十分钟。
下面是程序:
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#define N 100000
typedef struct node* link;
typedef struct {
float x;
float y;
} point;
struct node {
point p;
link next;
};
link **grid;
int G;
float d;
int cnt = 0;
int **malloc2d(int r, int c) {
int i;
int **t = malloc(r * sizeof(int *));
for (i = 0; i < r; i++)
t[i] = malloc(c * sizeof(int));
return t;
}
void grid_insert(float x, float y) {
int i, j;
link s;
int X = x * G + 1;
int Y = y * G + 1;
link t = malloc(sizeof *t);
t->p.x = x;
t->p.y = y;
for (i = X - 1; i <= X + 1; i++)
for (j = Y - 1; j <= Y + 1; j++)
for (s = grid[i][j]; s != NULL; s = s->next)
if (((s->p.x - t->p.x) * (s->p.x - t->p.x) + (s->p.y - t- >p.y) * (s->p.y - t->p.y)) < d * d)
cnt++;
t->next = grid[X][Y];
grid[X][Y] = t;
}
int main(int argc, char *argv[]) {
int i, j;
d = 0.01;
G = 1 / d;
grid = malloc2d(G + 2, G + 2);
for (i = 0; i < G + 2; i++)
for (j = 0; j < G + 2; j++)
grid[i][j] = NULL;
for (i = 0; i < N; i++)
grid_insert(1.0 * rand() / RAND_MAX, 1.0 * rand() / RAND_MAX);
printf("%d edges shorter than %f/n", cnt, d);
return 0;
}