2

我正在开发一个数组结构只是为了好玩。这种结构,由模板参数概括,在启动时预先分配给定数量的项目,然后,如果“忙碌”项目多于可用项目,函数将重新分配内部缓冲区。测试代码是:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

template <typename T> struct darray_t {
    size_t items;
    size_t busy;
    T     *data;
};

#define DARRAY_REALLOC_ITEMS_STEP 10

#define da_size(da) (da)->busy

template <typename T>
void da_init( darray_t<T> *da, size_t prealloc ){
    da->items = prealloc;
    da->busy  = 0;
    da->data  = (T *)malloc( sizeof(T) * prealloc );
}

template <typename T> T *da_next( darray_t<T> *da ){
    if( da->busy >= da->items ){
        da->data   = (T *)realloc( da->data, sizeof(T) * DARRAY_REALLOC_ITEMS_STEP );
        da->items += DARRAY_REALLOC_ITEMS_STEP;
    }
    return &da->data[ da->busy++ ];
}

int main(){
    darray_t<int> vi;
    int *n;

    da_init( &vi, 100 );

    for( int i = 0; i < 101; ++i ){
        n = da_next(&vi);
        *n = i;
    }

    for( int i = 0; i < da_size(&vi); ++i ){
        if( vi.data[i] != i ){
            printf( "!!! %d != %d\n", i, vi.data[i] );
        }
    }

    return 0;
}

正如你所看到的,我在开始时预先分配了 100 个整数指针,然后我一次重新分配了 10 个指针。在主函数中,我执行一个 for 循环来检查项目的完整性,如果一个数组项目不符合我的预期,我会打印它的值并且......你知道吗?我有以下消息:

!!!11 != 135121

事实上,索引 11 处的项目,应该是 '11',是 135121 !!!:S

如果我的代码不正确,你能告诉我吗?

谢谢

注意 我完全知道以这种方式混合 C 和 C++ 是丑陋的,而且我也知道如果使用这种结构会搞砸,例如:

darray_t<std::string>

这只是对 int 指针的测试。

4

2 回答 2

3

realloc 不会自动增加这块内存——你必须这样做。例如:

da->data=(T*)realloc(da->data, sizeof(T)*(da->items+DARRAY_REALLOC_ITEMS_STEP));

(你应该处理 realloc 返回 NULL)

于 2010-05-26T10:35:13.567 回答
1

块的大小不正确:

da->data   = (T *)realloc( da->data, sizeof(T) * DARRAY_REALLOC_ITEMS_STEP );

整个块与增量一样大。尝试

da->busy + sizeof(T) * DARRAY_REALLOC_ITEMS_STEP
于 2010-05-26T10:35:12.757 回答