在C ++中实现“最近点对”的问题

时间:2012-04-01 20:07:10

标签: c++ algorithm data-structures

我正在尝试根据Cormen书和维基百科的文章在C ++中实现最接近的一对点,我认为算法是正确的,但它只适用于非常小的数据。代码如下:

#include <cstdio>
#include <algorithm>
#include <cmath>
#define REP(i,n) for(int i=0;i<n;i++)

using namespace std;

struct point
{
    long long x, y;
};
struct dist
{
    long long x_1,y_1,x_2,y_2, distance;
} dis;

inline bool OrdX(const point &a, const point &b)
{
    if(a.x==b.x)
    {
        return a.y<b.y;
    }
    return a.x<b.x;
}

inline int OrdY(const point &a, const point &b)
{
    if(a.y==b.y)
    {
        return a.x<b.x;
    }
    return a.y<b.y;
}


// is - function that check is a an element of X_L array
inline bool is(const point &a, point *X_L, int p, int k)
{
    if(p<=k)
    {
        int center = (p+k)/2;

        if(X_L[center].x == a.x)
        {
            return true;
        }
        if(X_L[center].x > a.x)
        {
            return is(a, X_L, p, center-1);
        }
        else
        {
            return is(a, X_L, center+1, k);
        }
    }

    return false;
}


// odl - function takes two points and return distance between them ^2
inline long long odl(const point &a, const point &b)
{
    return ((a.x-b.x)*(a.x-b.x))+((a.y-b.y)*(a.y-b.y));
}


int tmp;

// fun - function that returns the pair of closest points using divide & conquer
struct dist fun(int n, point *X, point *Y)
{
    // if there are less that 4 points - it checks it using bruteforce
    if(n<4)
    {
        if(odl(X[0], X[1]) < dis.distance)
        {
            dis.distance = odl(X[0],X[1]);
            dis.x_1 = X[0].x;
            dis.y_1 = X[0].y;
            dis.x_2 = X[1].x;
            dis.y_2 = X[1].y;
        }

        if(n==3)
        {
            if(odl(X[0], X[2]) < dis.distance)
            {
                dis.distance = odl(X[0],X[2]);
                dis.x_1 = X[0].x;
                dis.y_1 = X[0].y;
                dis.x_2 = X[2].x;
                dis.y_2 = X[2].y;
            }
            if(odl(X[1], X[2]) < dis.distance)
            {
                dis.distance = odl(X[1],X[2]);
                dis.x_1 = X[1].x;
                dis.y_1 = X[1].y;
                dis.x_2 = X[2].x;
                dis.y_2 = X[2].y;
            }
        }
    }
    // otherwise it divides points into two arrays and runs fun 
    // recursively foreach part
    else
    {
        int p=n/2;

        int PPP = (X[p].x + X[p-1].x)/2;


        point *X_L = new point[p];
        point *X_R = new point[n-p];
        point *Y_L = new point[p];
        point *Y_R = new point[n-p];

        REP(i,p)
        X_L[i] = X[i];

        for(int r=p; r<n; r++)
        {
            X_R[r-p] = X[r];
        }

        int length_Y_L = 0;
        int length_Y_R = 0;

        REP(i,n)
        {
            if(is(Y[i], X_L, 0, p))
            {
                Y_L[length_Y_L++] = Y[i];
            }
            else
            {
                Y_R[length_Y_R++] = Y[i];
            }
        }


        dist D_L = fun(p, X_L, Y_L);
        dist D_R = fun(n-p, X_R, Y_R);
        dist D;

        if(D_L.distance < D_R.distance)
        {
            D = D_L;
        }
        else
        {
            D = D_R;
        }

        tmp = 0;
        point *Y2 = new point[n];

        double from = sqrt((double)D.distance);

        for(int r=0; r<n; r++)
        {
            if(Y[r].x > (long long)PPP-from && Y[r].x < (long long)PPP + from)
            {
                Y2[tmp++] = Y[r];
            }
        }

        //--tmp;
        //int xxx = min(7, tmp-r);
        int r = 0;
        for(int j=1; j<min(7, tmp-r); j++)
        {
            if(odl(Y2[r], Y2[r+j]) < D.distance)
            {
                D.distance = odl(Y2[r], Y2[r+j]);
                D.x_1 = Y2[r].x;
                D.y_1 = Y2[r].y;
                D.x_2 = Y2[r+j].x;
                D.y_2 = Y2[r+j].y;
            }
            r++;
        }

        dis = D;

    }
    return dis;
}

int main()
{
    int n;

    n = 7;

    point *X = new point[n];
    point *Y = new point[n];

    for(int i=0; i< 7; i++)
    {
        X[i].x = 0;
        X[i].y = 10*i;
    }

    /*
    REP(i,n)
    {
    scanf("%lld %lld", &X[i].x, &X[i].y);
    }
    */
    sort(X, X+n, OrdX);

    REP(i,n)
    Y[i] = X[i];

    sort(Y, Y+n, OrdY);

    dis.distance = odl(X[0], X[1]);

    dis.x_1 = X[0].x;
    dis.y_1 = X[0].y;
    dis.x_2 = X[1].x;
    dis.y_2 = X[1].y;

    dist wynik = fun(n, X, Y);

    printf(" %lld %lld\n %lld %lld\n", wynik.x_1, wynik.y_1, wynik.x_2, wynik.y_2);
    return 0;
}

我收到此错误:

 malloc.c:3096: sYSMALLOc: Assertion `(old_top == (((mbinptr) (((char
*) &((av)->bins[((1) - 1) * 2])) - __builtin_offsetof (struct
malloc_chunk, fd)))) && old_size == 0) || ((unsigned long) (old_size)
>= (unsigned long)((((__builtin_offsetof (struct malloc_chunk,
fd_nextsize))+((2 * (sizeof(size_t))) - 1)) & ~((2 * (sizeof(size_t)))
- 1))) && ((old_top)->size & 0x1) && ((unsigned long)old_end &
pagemask) == 0)' failed.

我已经尝试过loooking来解释这个错误,但找不到任何清楚的东西:/。 你能帮我解决一下吗?感谢

2 个答案:

答案 0 :(得分:2)

该消息表示您已使用动态分配的内存做了一些不好的事情。也许你将对象释放两次,或者在类似数组的动态分配对象的开头或结尾之外写入内存。

在Linux上,工具valgrind可能有助于确定程序执行中的第一个位置,它会成为一个嘘声。

顺便说一下,你的宏:

#define REP(i,n) for(int i=0;i<n;i++)

定义不明确。 n的替换应该括在括号中,因为n可能是一个对<运算符具有错误优先级的表达式。例如:REP(i, k < m ? z : w)。你想要:

#define REP(var,n) for(int var=0;var<(n);var++)

var提醒程序员这个参数是变量名,而不是任意表达式。

答案 1 :(得分:1)

您的功能is是多余的;这只是std::binary_search。这对@ sbi的可读性问题有很大帮助。

这样的块中也存在相当多的冗余
        dis.distance = odl(X[0],X[1]);
        dis.x_1 = X[0].x;
        dis.y_1 = X[0].y;
        dis.x_2 = X[1].x;
        dis.y_2 = X[1].y;

您可以为此编写一个简单的函数dist calcDist(point,point)。您应该将所有点定义和相关函数移动到单独的标题"point.h",再次使其可读。

至于内存问题:首先,数组X_LX_R并不是必需的。它们包含与X相同的数据,因此您可以分别指向&(X[0])&(X[p)Y_LY_R是随机播放的版本,因此您需要将数据复制到数组中。但是,如果您使用new[]分配它们,则您负责使用delete[]进行清理。您似乎只能使用std::vector<point> Y_L。无需簿记,vector为您做到了这一点。只需致电Y_L.push_back(Y[i])