迹忆客 专注技术分享

本文将介绍如何在 C++ 中从函数中返回数组。

在 C++ 中使用指针操作从函数中返回 C 式数组的方法

在 C/C++ 中，当 array[] 符号作为函数参数传递时，它只是一个指针，指向交给数组的第一个元素。因此，我们需要构造的函数原型是返回存储在数组中的类型的指针；在我们的例子中，它是 int。一旦函数返回，我们就可以用数组符号 [] 或 dereferencing 指针本身来访问它的元素（也许可以像之前那样做一些指针运算）。

#include <iostream>

using std::cout; 
using std::cin;
using std::endl; 
using std::string;

int *MultiplyArrayByTwo(int arr[], int size){
    for (int i = 0; i < size; ++i) {
        arr[i] *= 2;
    }
    return arr;
}

int main(){
    constexpr int size = 10;
    int c_array[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    int *ptr = MultiplyArrayByTwo(c_array, size);

    cout << "array = [ ";
    for (int i = 0; i < size; ++i) {
        cout << ptr[i] << ", ";
    }
    cout << "]" << endl;

    return EXIT_SUCCESS;
}

输出：

array = [ 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, ]

请注意，MultiplyArrayByTwo 函数可以实现为 void，我们可以从 c_array 变量中访问修改后的元素，因为它们指向同一个地址，如下例所示。

#include <iostream>

using std::cout;
using std::cin;
using std::endl
using std::string;

void MultiplyArrayByTwo(int arr[], int size)
{
    for (int i = 0; i < size; ++i) {
        arr[i] *= 2;
    }
}

int main(){
    constexpr int size = 10;
    int c_array[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    MultiplyArrayByTwo(c_array, size);

    cout << "array = [ ";
    for (int i = 0; i < size; ++i) {
        cout << c_array[i] << ", ";
    }
    cout << "]" << endl;

    return EXIT_SUCCESS;
}

在 C++ 中使用 vector 容器从函数中返回数组

在这个版本中，我们将数组存储在一个 vector 容器中，这个容器可以动态地扩展它的元素，所以不需要传递 size 参数。

注意，我们返回的是一个与函数不同的对象，所以我们不能使用 cpp_array 来访问它。我们需要创建一个新的变量，并将函数的返回值赋给它。然后我们就可以对其进行迭代，并输出结果。

#include <iostream>
#include <vector>

using std::cout;
using std::cin;
using std::endl
using std::string;
using std::vector;

vector<int> MultiplyArrayByTwo(const vector<int> *arr)
{
    vector<int> tmp{};

    for (const auto &item : *arr) {
        tmp.push_back(item * 2);
    }
    return tmp;
}

int main(){
    vector<int> cpp_array = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    auto cpp_array_2x = MultiplyArrayByTwo(&cpp_array);

    cout << "array = [ ";
    for (int i : cpp_array_2x) {
        cout << i << ", ";
    }
    cout << "]" << endl;

    return EXIT_SUCCESS;
}