C++中如何修改返回数组指针的函数以支持动态数组？

首先，我们来解决你提出的两个核心问题：修改test2使其支持动态数组，以及如何传递数组大小。

1. 让test2支持动态数组的两种可行方案 #

方案一：模板函数（编译期确定数组大小）

C++中，指向数组的指针类型是包含数组大小的（比如int(*)[3]代表指向含3个int的数组的指针）。如果想让函数支持任意固定大小的动态数组，最安全的方式是用模板把数组大小作为编译期参数，同时通过引用传递大小给调用者：

#include <iostream>
using namespace std;

// 模板函数，N为编译期确定的数组大小
template <size_t N>
int (*test2(size_t &sz))[N] {
    sz = N; // 通过引用将数组大小传递给调用者
    // 动态分配含N个int的数组，返回指向该数组的指针
    int (*out)[N] = new int[N];
    
    // 示例：给数组填充初始值1到N
    for (size_t i = 0; i < N; ++i) {
        (*out)[i] = i + 1;
    }
    
    return out;
}

int main() {
    size_t sz2;
    // 调用时指定数组大小为3，auto自动推导返回类型为int(*)[3]
    auto array2 = test2<3>(sz2);
    
    // 遍历输出数组元素
    for (size_t i = 0; i != sz2; ++i) {
        cout << (*array2)[i] << " ";
    }
    cout << endl;
    
    delete[] array2; // 必须用delete[]释放动态分配的数组，避免内存泄漏
    return 0;
}

方案二：运行时动态指定大小（更灵活的安全方案）

如果需要在运行时动态确定数组大小，C++标准不允许直接返回指向运行时大小数组的指针（因为数组大小是类型的一部分，必须是编译期常量）。这时候更推荐用std::unique_ptr<int[]>来管理动态数组，它会自动释放内存，同时同样可以通过引用传递大小：

#include <iostream>
#include <memory>
using namespace std;

std::unique_ptr<int[]> test2(size_t &sz) {
    sz = 5; // 这里可以替换为任意运行时确定的大小
    auto out = std::make_unique<int[]>(sz);
    
    // 示例：给数组填充初始值10到14
    for (size_t i = 0; i < sz; ++i) {
        out[i] = 10 + i;
    }
    
    return out;
}

int main() {
    size_t sz2;
    auto array2 = test2(sz2);
    
    // 遍历输出数组元素
    for (size_t i = 0; i != sz2; ++i) {
        cout << array2[i] << " ";
    }
    cout << endl;
    // 无需手动释放内存，unique_ptr会自动处理
    return 0;
}

2. 关于数组大小的传递 #

和你提供的test函数逻辑一致，我们可以通过引用参数（比如size_t &sz）将动态数组的大小传递给调用者，这样调用者就不需要硬编码数组长度，能灵活地遍历数组。

关键注意事项 #

• 用new[]分配的动态数组，必须对应使用delete[]释放，否则会造成内存泄漏；
• 模板方案适合大小固定的场景，数组大小在编译期确定；如果需要运行时动态调整大小，std::vector或std::unique_ptr<int[]>是更安全、更易维护的选择。

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