c - 如何在 C 函数中返回 int[N]*

Question

通过形参，我可以声明一个 int[N]* 类型的指针(在下面的代码中是 int (*A)[COL]) 。但我想知道如何将函数返回到这种类型的指针？

在我的代码中，int (*)[COL] 没有任何意义。

非常感谢你。

int (*)[COL] creat_2D_Array(int(*A)[COL], int ROW, int COL) {
    for (int i = 0; i < ROW; i++)
    {
        for (int j = 0; j < COL; j++)
        {
            A[i][j] = rand() % 20 + 1;
        }
    }
    return A;
}
//I add some code below after discussion with user3386109

//Or how can i return to a pointer of int[10]*
int (*)[10] creat_2D_Array(int(*A)[10]) {
    for (int i = 0; i < 5; i++)
    {
        for (int j = 0; j < 10; j++)
        {
            A[i][j] = rand() % 20 + 1;
        }
    }
    return A;
}

Answer 1

语法类似于函数指针，但您只需使用 [...] 而不是 (...)，例如:

type (*identifier)[size]

将函数名称和参数列表放入(* ... )中:

int (*creat_2D_Array(int (*A)[COL], int ROW, int COL_param))[COL] {
    for (int i = 0; i < ROW; i++)
    {
        for (int j = 0; j < COL; j++)
        {
            A[i][j] = rand() % 20 + 1;
        }
    }
    return A;
}

int (*creat_2D_Array_1(int(*A)[10]))[10] {
   for (int i = 0; i < 5; i++)
   {
       for (int j = 0; j < 10; j++)
       {
           A[i][j] = rand() % 20 + 1;
       }
   }
   return A;
}

第一个 creat_2D_Array 函数中的 COL 必须是常量表达式。

我有时使用 void* 指针从此类函数返回一个指向未知大小数组的指针，并在函数参数列表中组合简单的 VLA 声明，并期望调用者进行转换:

  void *creat_2D_Array(int ROW, int COL, int A[ROW][COL]) {
    for (int i = 0; i < ROW; i++)
    {
        for (int j = 0; j < COL; j++)
        {
            A[i][j] = rand() % 20 + 1;
        }
    }
    return A;
}

int main() {
   int input[5][6];
   int (*output)[6] = creat_2D_Array(5, 6, input);
   // remember these are pointers, you return the same pointer you inputted
   assert(input == output);
}

我发现这种方法很难维护。向某些模糊的对象或结构编写一些访问 API 和函数更容易使用(并允许断言和其他一些检查)。