写入函数的困难,它返回两个数组的交集

时间:2017-01-31 15:54:20

标签: c arrays dynamic-memory-allocation

我试图编写一个找到两个数组交集的函数

我只是不明白为什么它不能很好地运作。这是我的功能:

int* IntersectionOfArrays(int* arr1, int size1, int* arr2, int size2, int* sizeRes) 
{
    int* res=(int*)malloc(1*sizeof(int));  //res is the array of the resolution of intersection//
    int i = 0, j = 0;
    *sizeRes = 0;

    merge_sort(arr1,0, size1-1);  //sorting the arrays//
    merge_sort(arr2,0, size2-1);

    while (i < size1 && j < size2) 
    {
        if (arr1[i] < arr2[j])
            i++;
        else if (arr1[i] > arr2[j]) 
            j++;
        else 
        {   
        res[*sizeRes] = arr1[i]; //getting the same elements of the two arrays - the intersection values//
        i++;
        j++;
        (*sizeRes)++;
        res = (int*)realloc(res, 1*sizeof(int)); //allocating more memory as required - according to the size of res(intersection)//
        }   
    }

    if (*sizeRes==0)  //if the intersection is empty 
        return NULL;
    return res;
}

此函数编译,但由于我收到垃圾邮件,因此无法正常工作。 我想知道应该如何修复这个功能。

1 个答案:

答案 0 :(得分:2)

res = (int*)realloc(res, 1*sizeof(int)); 
//allocating more memory as required - according to the size of res(intersection)//

但与您的评论所暗示的相反,您并没有增加数组的大小。但相反,您再次为一个整数分配内存。请尝试以下方法:

res = realloc(res, (*sizeRes + 1) * sizeof(int));

此外,在使用realloc()

时使用临时指针
int* temp = realloc(res, (*sizeRes + 1) * sizeof(int));
if(temp == NULL) {
    //handle unsuccessful memory reallocation
}
else {
    res = temp;
}

通过这样做,即使重新分配失败,也可以使用res