100个小问题_每日一题_第2题

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; 
}