如何在C语言中返回矩阵对角线元素数组？新手内存分配报错求助

原代码中的核心错误： #

• 未定义变量与冗余逻辑：arr_len未声明就直接使用，而且方阵的对角线元素数量固定为n，完全不需要通过双层循环计数，属于纯冗余操作。
• 数组赋值错误：把目标数组ret_arr误写为arr_len[k]，arr_len是整型变量而非数组，直接访问会触发非法内存访问，导致段错误。
• 函数无返回值：return_diagonal声明返回float*类型，但函数末尾没有return ret_arr;语句，调用后得到的是随机无效指针。
• 内存泄漏：main函数中先给ret_arr分配内存，随后又被函数返回的指针覆盖，之前分配的内存无法释放，造成内存泄漏。
• 逻辑冗余：双层循环遍历找对角线完全没必要，利用行号等于列号的规律，单层循环即可完成访问。

正确实现代码 #

修正后的return_diagonal函数：

float* return_diagonal(float** matrix, int n) {
    // 直接分配n个float的内存，无需额外计数
    float *ret_arr = malloc(n * sizeof(float));
    if (ret_arr == NULL) { // 必须检查内存分配是否成功
        return NULL;
    }
    // 单层循环直接取对角线元素，无需内层遍历
    for (int i = 0; i < n; i++) {
        ret_arr[i] = matrix[i][i];
    }
    return ret_arr; // 必须返回分配好的数组指针
}

修正后的main调用逻辑：

float* ret_arr = return_diagonal(matrix, n);
if (ret_arr == NULL) { // 检查函数返回是否有效
    fprintf(stderr, "Memory allocation failed\n");
    return 1;
}
printf("Diagonal:\n");
for (int i = 0; i < n; i++) {
    printf("%.2f\n", ret_arr[i]); // 格式化输出更易读
}
free(ret_arr); // 用完必须释放动态分配的内存，避免泄漏

额外注意事项 #

• 始终检查malloc的返回值，避免内存分配失败后使用空指针。
• 动态分配的内存使用完毕后必须用free释放，防止内存泄漏。
• 方阵对角线元素的索引规律是行号等于列号，用单层循环即可完成遍历，无需双层循环浪费性能。

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