函数返回数组代码示例
时间: 2024-06-03 10:03:33 浏览: 14
当函数需要返回一个数组时,可以使用以下两种方法:
1 使用指针作为函数的返回值:函数可以返回一个指向数组的指针。这样做的好处是可以直接修改原始数组的值。以下是一个示例代码:
```c++
#include <iostream>
using namespace std;
int* createArray(int size) {
int* arr = new int[size];
for (int i = 0; i < size; i++) {
arr[i] = i + 1;
}
return arr;
}
int main() {
int size = 5;
int* arr = createArray(size);
for (int i = 0; i < size; i++) {
cout << arr[i] << " ";
}
delete[] arr; 记得释放内存
return 0;
}
```
2. 使用动态数组:函数可以返回一个动态分配的数组,这样可以避免手动释放内存。以下是一个示例代码:
```c++
#include <iostream>
#include <vector>
using namespace std;
vector<int> createArray(int size) {
vector<int> arr(size);
for (int i = 0; i < size; i++) {
arr[i] = i + 1;
}
return arr;
}
int main() {
int size = 5;
vector<int> arr = createArray(size);
for (int i = 0; i < size; i++) {
cout << arr[i] << " ";
}
return 0;
}
```