Como devolver um vetor a partir de uma função em C++
-
Utilize o
vector<T> func()Notação para retornar vetor a partir de uma função -
Utilize o
vector<T> &func()Notação para retornar vetor a partir de uma função
Este artigo apresentará como devolver um vetor de uma função de forma eficiente em C++.
Utilize o vector<T> func() Notação para retornar vetor a partir de uma função
O retorno por valor é o método preferido se devolvermos uma variável vector declarada na função. A eficiência deste método provém de seus movimentos-semânticos. Isso significa que retornar um vector não copia o objeto, evitando assim o desperdício de velocidade/espaço extra. Sob o capô, ele aponta o ponteiro para o objeto vector retornado, proporcionando assim um tempo de execução do programa mais rápido do que a cópia de toda a estrutura ou classe teria sido necessária.
#include <iostream>
#include <iterator>
#include <vector>
using std::cout;
using std::endl;
using std::vector;
vector<int> multiplyByFour(vector<int> &arr) {
vector<int> mult;
mult.reserve(arr.size());
for (const auto &i : arr) {
mult.push_back(i * 4);
}
return mult;
}
int main() {
vector<int> arr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
vector<int> arrby4;
arrby4 = multiplyByFour(arr);
cout << "arr - | ";
copy(arr.begin(), arr.end(), std::ostream_iterator<int>(cout, " | "));
cout << endl;
cout << "arrby4 - | ";
copy(arrby4.begin(), arrby4.end(), std::ostream_iterator<int>(cout, " | "));
cout << endl;
return EXIT_SUCCESS;
}
Resultado:
arr - | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
arrby4 - | 4 | 8 | 12 | 16 | 20 | 24 | 28 | 32 | 36 | 40 |
Utilize o vector<T> &func() Notação para retornar vetor a partir de uma função
Este método utiliza a notação de retorno por referência, que é mais adequada para retornar grandes estruturas e classes. Note que não retorna a referência da variável local declarada na própria função, pois ela leva a uma referência pendente. No exemplo a seguir, passamos o vetor arr por referência e o retornamos também como referência.
#include <iostream>
#include <iterator>
#include <vector>
using std::cout;
using std::endl;
using std::vector;
vector<int> &multiplyByFive(vector<int> &arr) {
for (auto &i : arr) {
i *= 5;
}
return arr;
}
int main() {
vector<int> arr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
vector<int> arrby5;
cout << "arr - | ";
copy(arr.begin(), arr.end(), std::ostream_iterator<int>(cout, " | "));
cout << endl;
arrby5 = multiplyByFive(arr);
cout << "arrby5 - | ";
copy(arrby5.begin(), arrby5.end(), std::ostream_iterator<int>(cout, " | "));
cout << endl;
return EXIT_SUCCESS;
}
Resultado:
arr - | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
arrby5 - | 5 | 10 | 15 | 20 | 25 | 30 | 35 | 40 | 45 | 50 |
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