Retornar array 2D de função em C++

Jinku Hu 12 outubro 2023
  1. Usar a Notação de Ponteiro para devolver um array 2D a partir de uma função em C++
  2. Utilizar notação Pointer to Pointer para devolver matriz 2D da função em C++
Retornar array 2D de função em C++

Este artigo irá introduzir como devolver um array 2D de uma função em C++.

Usar a Notação de Ponteiro para devolver um array 2D a partir de uma função em C++

A devolução pelo ponteiro é o método preferido para objectos maiores em vez de os devolver pelo valor. Uma vez que a matriz 2D pode ficar bastante grande, é melhor passar o ponteiro para o primeiro elemento do array, como demonstrado no seguinte exemplo de código. Note-se que o parâmetro para matriz 2D em ModifyArr é definido com a notação arr[][SIZE] para aceder aos seus elementos com parênteses no âmbito da função.

#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, ]

Utilizar notação Pointer to Pointer para devolver matriz 2D da função em C++

Como alternativa, podemos utilizar um ponteiro para a notação de ponteiro para devolver a matriz da função. Este método tem uma vantagem sobre outros se os objectos a serem devolvidos forem atribuídos dinamicamente. Normalmente, deve-se modificar a expressão de acesso ao elemento uma vez que o ponteiro é devolvido no âmbito do chamador. Note-se que lançamos o endereço do array para int* e depois desreferenciamos para obter os 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

Artigo relacionado - C++ Array