如何在 C 中对 malloc() 数组进行一行赋值？

Question

在 C 中，我可以像这样在堆栈上初始化一个数组：

SOME_DATA_TYPE* x = (SOME_DATA_TYPE[5]) {v1, v2, v3, v4, v5};

是否有类似的单行方法将值分配给堆上的malloc()-ed 数组？

SOME_DATA_TYPE* y = malloc(sizeof(SOME_DATA_TYPE)*5);  
// what comes here?  

还是我必须遍历数组并单独分配值？

Answer 1

关于“初始化”malloc() 的结果的第一个问题是分配可能会失败。这里y 被初始化为某个指针值。引用的数据仍然不确定。

#define  element_count  5
SOME_DATA_TYPE *y = malloc(sizeof *y * element_count);
if (y == NULL) Handle_OutOfMemory();

使用C11，代码可以使用复合文字来设置y指向的数据，而不是初始化。

memcpy(y, (SOME_DATA_TYPE[element_count]) {v1, v2, v3, v4, v5}, sizeof *y * element_count);

---

在不检查分配的情况下使用 direct 单行代码不会是健壮的编程。

// one-liner, but not robust code
SOME_DATA_TYPE *y = memcpy(malloc(sizeof *y * element_count), 
    (SOME_DATA_TYPE[element_count]) {v1, v2, v3, v4, v5}, sizeof *y * element_count);

---

注意代码使用sizeof *pointer_variable * element_count 而不是sizeof (pointer_variable_dereferenced_type) * element_count，因为它更易于编码、不易出错、更易于查看和维护。两种方法都行。

Answer 2

你可以像这样定义一个函数：

#include <stdarg.h>
#include <stdlib.h>

int * intMallocInit (int length, ...) {
   // Initialize Variable Arguments
   va_list args;
   va_start (args, length); 

    // Allocate Memory
    int * arr = malloc (sizeof(int) * length);
    if (arr == NULL) return NULL;

    // Initialize Array
    for (int i = 0; i < length; i++) {
        arr[i] = va_arg(args, int);
    }

    // Clean Up and Return
    va_end(args)
    return arr;
}

我们可以这样称呼它：

int len = 3;
int * myArray = intMallocInit (len, 4, 5, 2);
if (myArray == NULL) return 0; 

for (int i = 0; i < len; i++) {
    printf("%d,", myArray [i]);
}

输出： 4,5,2,

注意：我们也可以定义floatMallocInit()等

你也可以使用 void 指针，或者 enum& 开关，但我不会去那里。

一般功能的未经测试的尝试 我确信围绕 fhe 开关有一个聪明的工作，但这是我能想到的。

#include <stdarg.h>
#include <stdlib.h>
enum {INT, FLOAT /* etc...*/}

 int getSize(int type) {
     int typeSize = 0;
     switch (type) {
        case INT: 
             typeSize = sizeof (int);
             break;
        case FLOAT:
             typeSize = sizeof (float);
             break;
        // Etc...
        default:
             return -1;
    }
    return typeSize;
 }


void * mallocInit (int type, int length, ...) {
   // Initialize Variable Arguments
   va_list args;
   va_start (args, length); 

    // Get Size of type
    int typeSize = getSize(type);
    if (typeSize == -1) return NULL;

    // Allocate Memory
    void * arr = malloc (typeSize * length);
    if (arr == NULL) return NULL;

    // Initialize Array, Maybe va_copy can be used? No good reference
    for (int i = 0; i < length; i++) {
         switch(type) {
            case INT:
             arr[i] = va_args(args, int);
             break;
            // Etc..
        }
    }

    // Clean Up and Return
    va_end(args)
    return arr;
}

调用

float myArray = mallocInit (FLOAT, 3, 1, 5, 7);

Answer 3

在不偏离自然 C 内存管理理念的情况下，一种方法是使用类似memdup 的函数。由于它不是标准的，您可能必须自己实现它

SOME_DATA_TYPE *y = memdup(
  (SOME_DATA_TYPE[5]) {v1, v2, v3, v4, v5},
  5 * sizeof(SOME_DATA_TYPE));