c - 将两个给定数组交替组合成一个数组的函数

Question

我正在尝试用 C 编写一个函数，它接受两个字符数组，交替地将数组的元素组合成一个新数组，然后返回指向新数组的指针。

我下面的程序编译正常，但每次运行程序时都会产生不正确的、不同的输出。

如果有人能帮助指出我哪里出错了，我将不胜感激。

#include <stdio.h>
#include <stdlib.h>
char* altCombine(char[], char[], int, int);

int main(int argc, char** argv){
    char arr1[4] = {'e','f','g', 'h'};
    char arr2[4] = {'a','b','c','d'};
    int size1 = 4;
    int size2 = 4;
    char* altArr;
    altArr = altCombine(arr1, arr2, size1, size2);
    for(int i = 0; i < 8; i++){
        printf("%c \n", altArr[i]);
    }

    return 0;
}

char* altCombine(char arr1[], char arr2[], int size1, int size2){
    int i = 0, a = 0, b = 0;
    int totalSize = (size1 + size2);
    char* nArr = malloc(sizeof(char) * totalSize);
    char newArr[totalSize];
    while(i < totalSize){
        if(a < size1){
            newArr[i] = arr1[a];
            i++; 
            a++;    
        }
        if(b < size2){
            newArr[i] = arr2[b];
            i++;
            b++;
        } 
    }    
    nArr = &newArr[0];
    return nArr;    
}

Answer 1

这样做

  char* nArr = malloc(sizeof(char) * totalSize);

你走在正确的轨道上。您在堆上分配所需的空间，因此内存在显式释放之前一直有效。

这个

  char newArr[totalSize];

另一方面，在函数的堆栈上分配 newArr，所以它在函数返回的那一刻就消失了。

此外，因为您有 nArr，所以您不需要 newArr。

因此，为了最终确定您采用的方法，将所有 newArr 替换为 nArr 并删除此行

  nArr = &newArr[0];

你就完成了。

---

正如一些评论中提到的，您也可以使用

将 newArr 的内容复制到 nArr 指向的位置

  memcpy(nArr, newArr, sizeof newArr);

代替

  nArr = &newArr[0];

后面的语句不会复制数组的内容，而只是将 newArr 的第一个元素的地址分配给 指针 nArr。这对您没有帮助，因为该地址下的值将在函数返回时被释放(见上文)。

同时覆盖 nArr 的值会泄漏之前调用 malloc() 分配的内存。

---

与您的问题无关:所有这些 int 最好是 size_t。