Cómo devolver el array 2D de la función en C++

Jinku Hu 12 octubre 2023
  1. Usar la notación de puntero para devolver un array 2D de una función en C++
  2. Usar la notación de puntero a puntero para devolver el array 2D de la función en C++
Cómo devolver el array 2D de la función en C++

Este artículo introducirá cómo devolver un array 2D desde una función en C++.

Usar la notación de puntero para devolver un array 2D de una función en C++

El retorno por el puntero es el método preferido para los objetos más grandes en lugar de devolverlos por su valor. Como el array 2D puede ser bastante grande, es mejor pasar el puntero al primer elemento del array, como se demuestra en el siguiente ejemplo de código. Observe que el parámetro para el array 2D en ModifyArr está definido con la notación arr[][SIZE] para acceder a sus elementos con corchetes en el ámbito de la función.

#include <iomanip>
#include <iostream>
#include <vector>

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

constexpr int SIZE = 4;

int *ModifyArr(int arr[][SIZE], int len) {
  for (int i = 0; i < len; ++i) {
    for (int j = 0; j < len; ++j) {
      arr[i][j] *= 2;
    }
  }
  return reinterpret_cast<int *>(arr);
}

int main() {
  int c_array[SIZE][SIZE] = {
      {1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}};

  cout << "input array\n";
  for (auto &i : c_array) {
    cout << " [ ";
    for (int j : i) {
      cout << setw(2) << j << ", ";
    }
    cout << "]" << endl;
  }
  cout << endl;

  int *ptr = ModifyArr(c_array, SIZE);

  cout << "modified array\n";
  for (int i = 0; i < SIZE; ++i) {
    cout << " [ ";
    for (int j = 0; j < SIZE; ++j) {
      cout << setw(2) << *(ptr + (i * SIZE) + j) << ", ";
    }
    cout << "]" << endl;
  }
  cout << endl;

  return EXIT_SUCCESS;
}

Resultado:

input array
 [  1,  2,  3,  4, ]
 [  5,  6,  7,  8, ]
 [  9, 10, 11, 12, ]
 [ 13, 14, 15, 16, ]

modified array
 [  2,  4,  6,  8, ]
 [ 10, 12, 14, 16, ]
 [ 18, 20, 22, 24, ]
 [ 26, 28, 30, 32, ]

Usar la notación de puntero a puntero para devolver el array 2D de la función en C++

Como alternativa, podemos usar una notación de puntero a puntero para devolver el array de la función. Este método tiene una ventaja sobre otros si los objetos a devolver se asignan dinámicamente. Normalmente, se debe modificar la expresión de acceso al elemento una vez que el puntero es devuelto en el ámbito del llamador. Fíjese que lanzamos la dirección del array a int* y luego la dereferenciación para obtener los valores.

#include <iomanip>
#include <iostream>
#include <vector>

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

constexpr int SIZE = 4;

int **ModifyArr2(int *arr, int len) {
  for (int i = 0; i < len; ++i) {
    for (int j = 0; j < len; ++j) *(arr + (i * len) + j) *= 2;
  }
  return reinterpret_cast<int **>(arr);
}

int main() {
  int c_array[SIZE][SIZE] = {
      {1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}};

  cout << "input array\n";
  for (auto &i : c_array) {
    cout << " [ ";
    for (int j : i) {
      cout << setw(2) << j << ", ";
    }
    cout << "]" << endl;
  }
  cout << endl;

  int **ptr2 = ModifyArr2(c_array[0], SIZE);

  cout << "modified array\n";
  for (int i = 0; i < SIZE; ++i) {
    cout << " [ ";
    for (int j = 0; j < SIZE; ++j) {
      cout << setw(2) << *((int *)ptr2 + (i * SIZE) + j) << ", ";
    }
    cout << "]" << endl;
  }
  cout << endl;

  return EXIT_SUCCESS;
}

Resultado:

input array
 [  1,  2,  3,  4, ]
 [  5,  6,  7,  8, ]
 [  9, 10, 11, 12, ]
 [ 13, 14, 15, 16, ]

modified array
 [  2,  4,  6,  8, ]
 [ 10, 12, 14, 16, ]
 [ 18, 20, 22, 24, ]
 [ 26, 28, 30, 32, ]
Autor: Jinku Hu
Jinku Hu avatar Jinku Hu avatar

Founder of DelftStack.com. Jinku has worked in the robotics and automotive industries for over 8 years. He sharpened his coding skills when he needed to do the automatic testing, data collection from remote servers and report creation from the endurance test. He is from an electrical/electronics engineering background but has expanded his interest to embedded electronics, embedded programming and front-/back-end programming.

LinkedIn Facebook

Artículo relacionado - C++ Array