Question

我在文件中有一个矩阵，例如：

其中第一行表示方阵顺序。我正在使用以下代码读取文件并将其存储到向量中（为简单起见，我删除了所有if检查）：

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

int read_matrix_file(const char *fname, double *vector)
{
    /* Try to open file */
    FILE *fd = fopen(fname, "r");
    char line[BUFSIZ];
    fgets(line, sizeof line, fd);

    int n;
    sscanf(line, "%d", &n)

    vector = realloc(vector, n * n * sizeof *vector);
    memset(vector, 0, n * n * sizeof *vector);

    /* Reads the elements */
    int b;
    for(int i=0; i < n; i++) {
        // Read the i-th line into line
        if (fgets(line, sizeof line, fd) == NULL) {
            perror("fgets");
            return(-1);
        }

        /* Reads th j-th element of i-th line into the vector */
        char *elem_ptr = line;
        for (int j=0; j < n; j++) {
            if(sscanf(elem_ptr, "%lf%n", &vector[n*i+j] , &b) != 1) {
                perror("sscanf");
                return(1);
            }
            elem_ptr += b;
        }
    }
    fclose(fd);

    /* HERE PRINTS OK */
    for(int i=0; i<n*n; i++)
        printf("%i %f\n",i, vector[i]);

    return n;
}

read_matrix_file收到一个文件名和一个array的一个doubles并填充数组，返回矩阵顺序。可以在此代码块中看到预期的用法。

int main(void)
{

    const char *fname = "matrix.txt";
    double *vector = malloc(sizeof * vector);

    int n = read_matrix_file(fname, vector);

    /* Here prints junk */
    for(int i=0; i<n*n; i++)
        printf("%i %f\n",i, vector[i]);

    free(vector);
}

问题是，printf在read_matrix_file内部工作正常，但在main中似乎无效。

我正在函数外分配数组，并通过“引用”将其传递，但是我对realloc非常怀疑，很遗憾，我不知道如何解决或更好的方法。

Answer 1

您正在read_matrix_file()内重新分配内存，并将矩阵的元素存储在该内存区域中。但是，当您退出该函数时，由于指针vector是局部变量，因此当您离开该函数时，其新值将丢失。

当您回到main()内部时，向量仍然指向您先前用malloc()分配的（现在可能是无效的）存储区域。

您应该在调用read_matrix_file之前分配足够大的内存，或者如果要修改指针并查看更改反映在main()中，请传递双指针（**）

我的意思是这样的：

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

int read_matrix_file(const char *fname, double **p_vector)
{
    /* Try to open file */
    FILE *fd = fopen(fname, "r");
    char line[BUFSIZ];
    fgets(line, sizeof line, fd);

    int n;
    sscanf(line, "%d", &n);

    *p_vector = realloc(*p_vector, n * n * sizeof **p_vector);
    double *vector = *p_vector;

    memset(vector, 0, n * n * sizeof *vector);

    /* Reads the elements */
    int b;
    for(int i=0; i < n; i++) {
        // Read the i-th line into line
        if (fgets(line, sizeof line, fd) == NULL) {
            perror("fgets");
            return(-1);
        }

        /* Reads th j-th element of i-th line into the vector */
        char *elem_ptr = line;
        for (int j=0; j < n; j++) {
            if(sscanf(elem_ptr, "%lf%n", &vector[n*i+j] , &b) != 1) {
                perror("sscanf");
                return(1);
            }
            elem_ptr += b;
        }
    }
    fclose(fd);

    /* HERE PRINTS OK */
    for(int i=0; i<n*n; i++)
        printf("%i %f\n",i, vector[i]);

    return n;
}

主要通过以下方式调用它：

int n = read_matrix_file(fname, &vector);

编辑：请注意，此代码无法正确处理realloc()的失败。

数组元素在函数外“丢失”

1 个答案: