C语言计数排序中malloc分配的count_array内存释放问题求助

先解决栈数组的段错误问题 #

你用int count_array[MAX_VALUE];出现段错误，大概率是MAX_VALUE太大，栈空间不够导致栈溢出。C语言的栈空间通常只有几MB（比如8MB），如果MAX_VALUE设成1e6，int占4字节的话，这个数组就占4MB，再加上其他栈变量很容易超过栈的限制，直接触发段错误。

解决办法：放弃大的栈数组，老实用malloc在堆上分配内存。

再解决malloc/free的段错误问题 #

你用malloc后释放出现段错误，基本逃不开这几个原因，逐个排查：

1. 计数数组越界访问，破坏了堆结构 #

这是最常见的原因，比如：

• 数组里有负数，你直接用arr[i]作为count_array的索引，访问了负数下标，直接越界；
• 计算max的时候出错（比如数组全是负数，max还是初始的arr[0]，导致count_array大小是max+1，而max是负数，数组大小为0或负数，malloc出问题）；
• count_array的大小算错，比如少加了1（应该是max+1，你写成max），导致访问count_array[max]时越界。

解决：
如果数组包含负数，要同时找min和max，用偏移量计算索引：

• 计数数组大小 = max - min + 1；
• 统计元素时，用arr[i] - min作为索引；
• 写回原数组时，用i + min还原元素值。

2. 重复释放或释放无效指针 #

比如：

• 你在count_sort函数里已经free了count_array，外部又拿着这个指针去free；
• 指针被覆盖（比如函数里把count_array赋值给了其他变量，然后free了两次）；
• 释放了不是malloc分配的内存（比如把栈数组的指针传给free）。

正确的释放时机：
如果count_array只在count_sort函数内部使用，就在函数末尾、用完数组之后立即释放，不要把这个指针传出函数，避免外部误操作。

3. 没检查malloc的返回值 #

malloc可能分配失败（比如内存不足），返回NULL，这时候访问count_array会直接触发段错误。必须加判断。

完整的正确示例代码 #

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

void count_sort(int arr[], int n) {
    if (n <= 0) return;

    // 找出数组的最小、最大值
    int min = arr[0], max = arr[0];
    for (int i = 1; i < n; i++) {
        if (arr[i] < min) min = arr[i];
        if (arr[i] > max) max = arr[i];
    }

    // 计算计数数组的大小，分配内存
    int count_size = max - min + 1;
    int *count_array = (int*)malloc(count_size * sizeof(int));
    if (count_array == NULL) {
        perror("malloc failed"); // 打印内存分配失败原因
        exit(EXIT_FAILURE);
    }

    // 初始化计数数组为0
    memset(count_array, 0, count_size * sizeof(int));

    // 统计每个元素出现次数
    for (int i = 0; i < n; i++) {
        int idx = arr[i] - min; // 偏移处理负数
        count_array[idx]++;
    }

    // 将排序结果写回原数组
    int arr_idx = 0;
    for (int i = 0; i < count_size; i++) {
        while (count_array[i] > 0) {
            arr[arr_idx++] = i + min;
            count_array[i]--;
        }
    }

    // 用完立即释放，这里是最安全的时机
    free(count_array);
}

// 测试用例
int main() {
    int arr[] = {5, 2, 9, 1, 5, 6, -3};
    int n = sizeof(arr)/sizeof(arr[0]);

    count_sort(arr, n);

    printf("排序后数组：");
    for (int i = 0; i < n; i++) {
        printf("%d ", arr[i]);
    }
    printf("\n");

    return 0;
}

新手避坑总结 #

• 不要用超过栈空间的大数组，堆分配（malloc）才是处理大数据量的正确方式；
• 计数排序一定要处理负数，否则必越界；
• 堆内存分配后，用完就释放，尽量在同一个函数内完成分配和释放，减少出错概率；
• 永远要检查malloc的返回值，避免空指针访问。

内容的提问来源于stack exchange，提问作者weito